"fsdp" Jobs

27 open tech roles matching “fsdp”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: PyTorch, DeepSpeed, FSDP. Every listing is re-checked daily and closed roles are removed.

Showing 20 of 27 results

Databricks

Join Databricks as a Staff Software Engineer to drive the architecture of a managed GPU training platform for large-scale AI models.

Databricks Mountain View, California; San Francisco, California $190k–$265k/yr Published 3 months ago
Kodiak Robotics

Join Kodiak Robotics as a Senior AI Infrastructure Engineer to optimize model training for autonomous technology.

Kodiak Robotics Mountain View, CA $190k–$260k/yr Published 2 months ago
Flexible on stack
Databricks

Join Databricks as a Senior Software Engineer to build and scale a managed GPU training platform for AI models.

Databricks Mountain View, California; San Francisco, California $160k–$225k/yr Published 3 months ago
Flexible on stack 70% coding
Fireworks AI

Design and optimize infrastructure for large-scale AI model training at a leading generative AI company.

Fireworks AI San Mateo Published 1 month ago
Flexible on stack
krea.ai

Join Krea as an ML Researcher to train diffusion models for image and video generation in a creative AI-focused environment.

krea.ai San Francisco Published 1 week ago
Flexible on stack
DeepL

Lead research on fine-tuning and steerability of LLM-based translation models in a collaborative AI-focused environment.

DeepL London Published 1 month ago
Flexible on stack
Hedra

Join Hedra as a Research Engineer to lead the development of action-conditioned world models in a pioneering Physical AI team.

Hedra San Francisco Published 5 months ago
Flexible on stack
Databricks

Lead a high-performing engineering team at Databricks to enhance the AI Runtime product for GPU training infrastructure.

Databricks Mountain View, California; San Francisco, California $228.6k–$297.1k/yr Published 2 months ago
Flexible on stack Heavy meetings
baseten

Join Baseten as a Software Engineer to architect and develop scalable infrastructure for ML training platforms.

baseten San Francisco Published 1 year ago
Flexible on stack
krea.ai

Join Krea as an ML Researcher to finetune diffusion models and enhance AI creative tools in a collaborative environment.

krea.ai San Francisco Published 1 week ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Machine Learning Research Engineer intern to enhance search quality through advanced deep learning techniques.

Perplexity AI Berlin Published 1 week ago
Periodic Labs

Join Periodic Labs as an ML Systems Engineer to build and optimize large-scale training and reinforcement learning infrastructure.

Periodic Labs Menlo Park, CA $250k–$350k/yr Published 4 months ago
Flexible on stack
baseten

Join Baseten as a senior software engineer to develop cutting-edge AI training products and enhance user workflows.

baseten San Francisco Published 7 months ago
Flexible on stack
Together AI

Join Together AI as a Research Engineer to optimize large-scale training infrastructure for cutting-edge AI models.

Together AI San Francisco $200k–$290k/yr Published 1 month ago
Flexible on stack
Cantina

Join Cantina as a Machine Learning Engineer to develop cutting-edge speech and audio generation systems in a collaborative environment.

Cantina Remote (U.S. or Europe) $200k–$220k/yr Published 1 month ago
Flexible on stack 70% coding
Hippocratic AI

Join Hippocratic AI as an Applied Scientist to enhance AI safety and clinical reasoning through reinforcement learning.

Hippocratic AI Menlo Park, CA Published 1 month ago
Flexible on stack
Mirelo AI

Join Mirelo AI as a Research Scientist to develop cutting-edge multimodal models for audio generation in a rapidly growing company.

Mirelo AI Berlin Published 9 months ago
Flexible on stack 70% coding
Reflection AI

Join Reflection AI as a Research Software Engineer to bridge research and production in cutting-edge AI training systems.

Reflection AI New York, NY Published 6 months ago
Flexible on stack
Cantina

Join Cantina as a Machine Learning Engineer to build advanced speech systems and contribute to innovative AI technology.

Cantina Remote (U.S. or Europe) $200k–$220k/yr Published 1 month ago
Flexible on stack 70% coding