"fsdp" Jobs

11 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 11 of 11 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
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
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
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
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
Twelve Labs

Lead the development of next-generation multimodal models at Twelve Labs, impacting thousands of customers worldwide.

Twelve Labs Seoul, South Korea Published 1 week ago
Flexible on stack
Coreweave

Drive new business opportunities in AI infrastructure as a Principal Solution Specialist at CoreWeave.

Coreweave San Francisco, CA / Seattle, WA $198k–$264k/yr Published 3 weeks ago
Flexible on stack