"pytorch" Jobs

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

Showing 20 of 319 results

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
Kodiak Robotics

Lead the architecture direction for multimodal transformers at Kodiak Robotics, focusing on AI-powered autonomous technology.

Kodiak Robotics Mountain View, CA $230k–$300k/yr Published 1 month ago
Coreweave

Join CoreWeave as a Forward Deployed Engineer to help customers build and deploy AI agents using the W&B Weave toolkit.

Coreweave Sunnyvale, CA / San Francisco, CA $182k–$242k/yr Published 1 month ago
Flexible on stack 70% coding
Instacart

Lead the design and development of core ML models for Instacart’s ads ecosystem in a fully remote role.

Instacart United States - Remote $201k–$253.5k/yr Published 4 months ago
Flexible on stack
Periodic Labs

Join Periodic Labs as a Computational Scientist to develop differentiable simulations for continuum-physics problems using AI.

Periodic Labs Menlo Park, CA $250k–$350k/yr Published 3 weeks ago
Flexible on stack
Fireworks AI

Drive growth and adoption for Fireworks AI's product portfolio as a Product Manager focused on developer experience.

Fireworks AI San Mateo Published 4 days ago
Cognition

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

Cognition San Francisco Published 1 month ago
AI-first team
Preference Model

Join Preference Model as a Senior Machine Learning Engineer to design RL environments for advancing ML capabilities.

Preference Model San Francisco Published 2 weeks ago
Flexible on stack
Inferact

Join Inferact as an AMD GPU performance engineer to optimize vLLM for the AMD accelerator ecosystem.

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

Join Reddit as a Staff Machine Learning Engineer to enhance ML-powered ad ranking systems in a fully remote role.

Reddit Remote - United States $230k–$322k/yr Published 3 days ago
Flexible on stack
Fireworks AI

Lead partnerships to create integrated AI factory solutions with a focus on OEM and infrastructure alliances.

Fireworks AI San Francisco Published 3 days ago
Fireworks AI

Lead the go-to-market strategy for Fireworks' strategic partner ecosystem, driving measurable business growth through collaboration.

Fireworks AI San Mateo Published 1 month ago
Worldcoin

Own face machine learning projects and improve biometric identification models in a collaborative AI team.

Worldcoin Munich Published 2 months 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 3 weeks 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
Fireworks AI

Join Fireworks AI as a GTM Enablement Lead to develop and deliver training programs that enhance sales efficiency in a high-growth environment.

Fireworks AI United States, Remote Published 1 month ago
Reddit

Build and optimize large-scale machine learning systems for recommendation and personalization at Reddit.

Reddit Remote - United States $190.8k–$267.1k/yr Published 1 month ago
Flexible on stack
Iambic Therapeutics

Join Iambic Therapeutics as a Machine Learning Scientist to innovate AI-based drug discovery with multimodal models.

Iambic Therapeutics Boston Office Published 1 month ago
Flexible on stack
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 2 months ago
Flexible on stack