Jax Jobs
159 open roles, updated daily
Filters2

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.
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 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.