"pytorch distributed" Jobs

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

Showing 20 of 80 results

Preference Model

Join Preference Model as a Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.

Preference Model San Francisco Published 1 week 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 2 weeks ago
Flexible on stack
Xaira Therapeutics

Join Xaira Therapeutics as an AI Scientist to innovate in computational protein design using advanced deep learning methods.

Xaira Therapeutics South San Francisco, California, United States $120k–$240k/yr Published 11 months ago
Flexible on stack
Goodfire

Join Goodfire as a Machine Learning Engineer to build interpretable AI systems with a world-class team.

Goodfire San Francisco, CA & New York, NY $200k–$400k/yr Published 9 months ago
P-1 AI

Join P-1 AI as an AI Research Scientist to advance spatial reasoning and physical world modeling in cutting-edge AI systems.

P-1 AI San Francisco Bay Area Published 1 month ago
Flexible on stack
Abridge

Join Abridge as a Machine Learning Infrastructure Engineer to optimize AI model inference infrastructure in a fast-paced healthcare startup.

Abridge SF Office Published 1 year ago
Flexible on stack
P-1 AI

Join P-1 AI as an AI Research Scientist to push the boundaries of AI applied to the physical world in a hands-on role.

P-1 AI San Francisco Bay Area Published 1 month ago
Flexible on stack
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
Lyft

Join Lyft as a Machine Learning Engineer to design and deploy cutting-edge ML systems in a collaborative environment.

Lyft San Francisco, CA $140.8k–$176k/yr Published 1 month ago
Flexible on stack 70% coding
Anthropic

Join Anthropic as a Staff Software Engineer to build scalable ML infrastructure for AI safety systems.

Anthropic San Francisco, CA $320k–$485k/yr Published 1 week ago
Flexible on stack
Inferact

Lead the engineering organization at Inferact to develop systems for vLLM, focusing on GPU performance and ML systems optimization.

Inferact San Francisco Published 1 month ago
Cognition

Join a small, highly selective team to enhance AI model capabilities at Cognition in San Francisco.

Cognition San Francisco, United States Published 1 month ago
AI-first team
Goodfire

Join Goodfire as the Head of Engineering to lead the development of AI interpretability tools with a world-class team.

Goodfire San Francisco, CA $300k–$450k/yr Published 8 months ago
Flexible on stack
Braze

Join Braze as a Staff Applied Scientist to lead transformative ML initiatives for personalized customer engagement.

Braze San Francisco $184k–$299.8k/yr Published 2 weeks ago
Flexible on stack
Neuralink

Join Neuralink's BCI team as a Machine Learning Engineer to develop cutting-edge neural interface solutions.

Neuralink Austin, Texas, United States; South San Francisco, California, United States $199k–$331k/yr Published 3 years ago
Flexible on stack
baseten

Join Baseten as a lead Software Engineer to own and develop production-grade Voice AI systems that impact daily lives.

baseten San Francisco Published 5 months 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 1 month ago
Flexible on stack
Anthropic
Anthropic New York City, NY; New York City, NY | Seattle, WA; San Francisco, CA $350k–$500k/yr Published 11 months ago
Handshake

Join Handshake as a Senior Software Engineer to build scalable ML infrastructure for a fast-growing AI data business.

Handshake San Francisco, CA Published 2 months ago
Flexible on stack