JAX Jobs
191 open roles, updated daily
Filters1
Apply for the Data Infra - Signals and Evaluations role at Matter Intelligence in San Francisco. Build datasets, benchmarks, and evaluation workflows.
Apply for the Data Infra - Signals and Evaluations role at Matter Intelligence in San Francisco. Build datasets, benchmarks, and evaluation workflows.
Apply for the Staff Machine Learning Engineer role at Phantom. Remote position involving growth ML initiatives, system architecture, and team mentorship.
Apply for the Staff Machine Learning Engineer role at Phantom. Remote position involving growth ML initiatives, system architecture, and team mentorship.
Apply for the ML Infrastructure Engineer role at Preference-Model in San Francisco. Build LLM post-training systems. Salary: $200K–$350K + equity.
Apply for the ML Infrastructure Engineer role at Preference-Model in San Francisco. Build LLM post-training systems. Salary: $200K–$350K + equity.
Join Preference-Model in San Francisco as a Technical Staff Research Engineer. Earn $200K-$350K plus equity, with visa sponsorship available.
Join Preference-Model in San Francisco as a Technical Staff Research Engineer. Earn $200K-$350K plus equity, with visa sponsorship available.
Apply for the Member of Technical Staff - Machine Learning Capabilities role at Preference-Model in San Francisco. Competitive salary and benefits.
Apply for the Member of Technical Staff - Machine Learning Capabilities role at Preference-Model in San Francisco. Competitive salary and benefits.
Join Discord's Safety ML team in the SF Bay Area. Build ML systems for content understanding and user safety. Base salary $220,000 to $275,000.
Join Discord's Safety ML team in the SF Bay Area. Build ML systems for content understanding and user safety. Base salary $220,000 to $275,000.
Apply for the Member of Technical Staff - Machine Learning role at inductive-bio in New York City, San Francisco, or Boston. Salary $180K–$250K + equity.
Apply for the Member of Technical Staff - Machine Learning role at inductive-bio in New York City, San Francisco, or Boston. Salary $180K–$250K + equity.
Apply for the Machine Learning Engineer role at DeepJudge in Zurich HQ. Build and scale enterprise AI systems for legal search products.
Apply for the Machine Learning Engineer role at DeepJudge in Zurich HQ. Build and scale enterprise AI systems for legal search products.
Apply for the 2027 Internship Onboard Infrastructure Engineer, ML Inference role at bedrock-robotics in San Francisco, CA.
Apply for the 2027 Internship Onboard Infrastructure Engineer, ML Inference role at bedrock-robotics in San Francisco, CA.
Join Waymo as a Senior ML Engineer in Predictive Planning. Build autonomous driving models with Python, C++, and deep learning. Salary $213k-$263k.
Join Waymo as a Senior ML Engineer in Predictive Planning. Build autonomous driving models with Python, C++, and deep learning. Salary $213k-$263k.
Apply for the 2027 Summer Intern PhD Vehicle Intent and Prediction role at Waymo in Mountain View, CA. Hourly pay of $85 USD.
Apply for the 2027 Summer Intern PhD Vehicle Intent and Prediction role at Waymo in Mountain View, CA. Hourly pay of $85 USD.
Apply for the Compute Solution Architect role at Mistral AI in New York. Support strategic AI infrastructure customers and scale GPU workloads.
Apply for the Compute Solution Architect role at Mistral AI in New York. Support strategic AI infrastructure customers and scale GPU workloads.
Join Anthropic in San Francisco as a Staff+ Software Engineer. Build ML inference infrastructure for safety systems. Salary: $320k-$485k.
Join Anthropic in San Francisco as a Staff+ Software Engineer. Build ML inference infrastructure for safety systems. Salary: $320k-$485k.
Join True Anomaly in Denver or Long Beach as a Machine Learning Engineer. Build autonomous space tech with a competitive salary and robust benefits.
Join True Anomaly in Denver or Long Beach as a Machine Learning Engineer. Build autonomous space tech with a competitive salary and robust benefits.
Apply for the Research Software Engineer, Post Training role at Thinking Machines in San Francisco. Salary: $350k-$475k. Remote options available.
Apply for the Research Software Engineer, Post Training role at Thinking Machines in San Francisco. Salary: $350k-$475k. Remote options available.
Apply for the Thermo ML Resident role at Extropic in Boston. Flexible part-time or full-time machine learning residency paying $75K–$200K.
Apply for the Thermo ML Resident role at Extropic in Boston. Flexible part-time or full-time machine learning residency paying $75K–$200K.
Join Extropic as a Thermo ML Resident in San Francisco. Flexible part-time or full-time role with $75K-$200K salary.
Join Extropic as a Thermo ML Resident in San Francisco. Flexible part-time or full-time role with $75K-$200K salary.
Apply for the Research Scientist, Map Scalability role at Waymo in Mountain View, San Francisco, or New York City. Salary $213,000 - $263,000 USD.
Apply for the Research Scientist, Map Scalability role at Waymo in Mountain View, San Francisco, or New York City. Salary $213,000 - $263,000 USD.
Apply for the Member of Technical Staff, Research PhD New Grad role at Fireworks AI in San Mateo or remotely. Earn $200k-$300k yearly.
Apply for the Member of Technical Staff, Research PhD New Grad role at Fireworks AI in San Mateo or remotely. Earn $200k-$300k yearly.
Join Isomorphic Labs in London as a Research Engineer for LLM Performance. Scale foundational AI models for drug discovery using JAX, PyTorch, and distributed training.
Join Isomorphic Labs in London as a Research Engineer for LLM Performance. Scale foundational AI models for drug discovery using JAX, PyTorch, and distributed training.
Join Maincode in Melbourne as a Forward-Deployed AI Engineer. Deploy local AI stacks for justice and community safety customers. Apply today.
Join Maincode in Melbourne as a Forward-Deployed AI Engineer. Deploy local AI stacks for justice and community safety customers. Apply today.
Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.
Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.
Join Mistral AI in Paris as a Research Engineer, Machine Learning. Build and optimize large-scale learning systems for open-weight models.
Join Mistral AI in Paris as a Research Engineer, Machine Learning. Build and optimize large-scale learning systems for open-weight models.