"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

Databricks

Lead a multidisciplinary research team to advance large-scale machine learning efficiency at Databricks.

Databricks Mountain View, California; San Francisco, California $270k–$340k/yr Published 4 months ago
Flexible on stack 60% coding
Twelve Labs

Drive research on Pegasus's complex problems in a hybrid role at a growing AI company focused on video understanding.

Twelve Labs Seoul, South Korea Published 5 months ago
Flexible on stack
Reflection AI

Join Reflection AI as a Member of Technical Staff to research and build solutions for large language models in a fast-paced startup environment.

Reflection AI San Francisco, CA Published 11 months ago
Flexible on stack
Anthropic
Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 12 months ago
Kodiak Robotics

Join Kodiak Robotics as a Staff Machine Learning Engineer to design and deploy ML systems for autonomous trucking.

Kodiak Robotics San Francisco Bay Area $200k–$265k/yr Published 5 months ago
Flexible on stack
Databricks
Databricks Bellevue, Washington; San Francisco, California $170.4k–$255.6k/yr Published 1 year ago
Databricks
Databricks San Francisco, California $280k–$350k/yr Published 5 months ago
baseten

Join Baseten as a Forward Deployed Engineer to solve complex AI challenges for leading companies.

baseten San Francisco Published 1 month ago
Flexible on stack
Together AI

Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to innovative model shaping projects.

Together AI San Francisco, Amsterdam $58–$70/hr Published 2 days 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
Together AI

Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to innovative model shaping projects.

Together AI San Francisco, Amsterdam $58–$70/hr Published 2 days ago
Flexible on stack
Databricks

Join Databricks as a Solutions Architect to shape big data solutions for leading tech companies with a focus on AI and data engineering.

Databricks Remote - California; San Francisco, California $180k–$247.5k/yr Published 1 month ago
Flexible on stack
Genmo

Join Genmo as a Research Scientist to lead initiatives in alignment and post-training for large-scale video generation models.

Genmo San Francisco HQ Published 6 days ago
Flexible on stack
Inferact

Join Inferact as a Developer Relations Engineer to shape how developers learn and build with vLLM, the AI inference engine.

Inferact San Francisco $200k–$400k/yr Published 3 months ago
Flexible on stack
Genmo

Join Genmo as a Research Engineer to advance visual generative AI in a fast-paced startup environment.

Genmo San Francisco HQ Published 6 days 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 8 months 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 2 weeks ago
Flexible on stack
Xaira Therapeutics

Join Xaira Therapeutics as an AI Scientist Intern to advance AI models for protein and antibody therapeutic design.

Xaira Therapeutics Seattle, Washington, United States; South San Francisco, California, United States $10k–$15k/mo Published 3 weeks ago
Flexible on stack
Genmo

Join Genmo as a Research Scientist to innovate in diffusion models for text-to-video generation.

Genmo San Francisco HQ Published 6 days ago
Flexible on stack