JAX Jobs
105 open roles, updated daily
Filters3
Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.
Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.
Apply for the AI Research Member of Technical Staff role at psi in Boston. Build AI agents and training systems to discover physics at scale.
Apply for the AI Research Member of Technical Staff role at psi in Boston. Build AI agents and training systems to discover physics at scale.
Apply for Member of Technical Staff, Applied AI at psi in Boston. Build AI systems to discover new physics at scale. Check requirements and apply.
Apply for Member of Technical Staff, Applied AI at psi in Boston. Build AI systems to discover new physics at scale. Check requirements and apply.

Apply for the LLM Scientist, AI for Science role at Mistral in Paris. Develop LLMs and deep learning frameworks for biology, chemistry, and physics.

Apply for the LLM Scientist, AI for Science role at Mistral in Paris. Develop LLMs and deep learning frameworks for biology, chemistry, and physics.

Apply for the Machine Learning Platform Engineer role at bjakcareer in Seoul, Korea. Build ML infrastructure, platforms, and production AI systems.

Apply for the Machine Learning Platform Engineer role at bjakcareer in Seoul, Korea. Build ML infrastructure, platforms, and production AI systems.

Apply for the Member of Technical Staff, Machine Learning role at bjakcareer in Poland. Build production ML systems, Python, PyTorch, JAX.

Apply for the Member of Technical Staff, Machine Learning role at bjakcareer in Poland. Build production ML systems, Python, PyTorch, JAX.

Join ACTAI as a Senior Machine Learning Engineer in Spain. Build core ML systems and own production-ready AI applications end-to-end.

Join ACTAI as a Senior Machine Learning Engineer in Spain. Build core ML systems and own production-ready AI applications end-to-end.

Apply for the Senior Machine Learning Engineer role at ACTAI in Poland. Build core ML systems for proactive AI products.

Apply for the Senior Machine Learning Engineer role at ACTAI in Poland. Build core ML systems for proactive AI products.

Join ActAI in Germany as a Machine Learning Platform Engineer to build and operate scalable ML infrastructure, serving systems, and production pipelines.

Join ActAI in Germany as a Machine Learning Platform Engineer to build and operate scalable ML infrastructure, serving systems, and production pipelines.

Build core ML components for production systems at ACTAI in Zurich, Switzerland. Focus on fine-tuning, evaluation, and real-world task completion.

Build core ML components for production systems at ACTAI in Zurich, Switzerland. Focus on fine-tuning, evaluation, and real-world task completion.

Join bjakcareer in Zurich as a Senior Machine Learning Engineer. Own production ML systems and build proactive AI applications.

Join bjakcareer in Zurich as a Senior Machine Learning Engineer. Own production ML systems and build proactive AI applications.

Apply for the Machine Learning Platform Engineer role at bjakcareer in Portugal. Build scalable ML infrastructure and AI production systems.

Apply for the Machine Learning Platform Engineer role at bjakcareer in Portugal. Build scalable ML infrastructure and AI production systems.

Apply for the Machine Learning Platform Engineer role at bjakcareer in Sweden. Build scalable AI infrastructure and production ML systems.

Apply for the Machine Learning Platform Engineer role at bjakcareer in Sweden. Build scalable AI infrastructure and production ML systems.

Build ML infrastructure and serving systems for ActAI in the United Kingdom. Apply on the official careers page.

Build ML infrastructure and serving systems for ActAI in the United Kingdom. Apply on the official careers page.

Build ML infrastructure and serving systems for ActAI in China. Full-time role focused on scalable AI platforms, distributed systems, and MLOps.

Build ML infrastructure and serving systems for ActAI in China. Full-time role focused on scalable AI platforms, distributed systems, and MLOps.

Join ACTAI in the United States as a Senior Machine Learning Engineer to build proactive AI systems. Apply on the official careers page.

Join ACTAI in the United States as a Senior Machine Learning Engineer to build proactive AI systems. Apply on the official careers page.

Join ACTAI as a Member of Technical Staff, Machine Learning in the United States. Build core ML components and production systems.

Join ACTAI as a Member of Technical Staff, Machine Learning in the United States. Build core ML components and production systems.

Apply for Senior Machine Learning Engineer at bjakcareer in China. Build proactive AI apps and own critical ML subsystems in production.

Apply for Senior Machine Learning Engineer at bjakcareer in China. Build proactive AI apps and own critical ML subsystems in production.

Apply for the Senior Machine Learning Engineer role at bjakcareer in the United Kingdom. Build core ML systems for proactive AI products.

Apply for the Senior Machine Learning Engineer role at bjakcareer in the United Kingdom. Build core ML systems for proactive AI products.

Apply for the Senior Machine Learning Engineer role at bjakcareer in Singapore. Own critical ML subsystems in production and build proactive AI products.

Apply for the Senior Machine Learning Engineer role at bjakcareer in Singapore. Own critical ML subsystems in production and build proactive AI products.

Apply for the Senior Machine Learning Engineer role at ACTAI in Seoul, Korea. Build core ML systems for proactive AI products.

Apply for the Senior Machine Learning Engineer role at ACTAI in Seoul, Korea. Build core ML systems for proactive AI products.

Build core ML components and production systems in Singapore with bjakcareer. Apply on the official careers page.

Build core ML components and production systems in Singapore with bjakcareer. Apply on the official careers page.

Build core ML components for production systems at ACTAI in Seoul, Korea. Focus on real-world machine learning, model fine-tuning, and scalable inference.

Build core ML components for production systems at ACTAI in Seoul, Korea. Focus on real-world machine learning, model fine-tuning, and scalable inference.

Apply for the Member of Technical Staff, Machine Learning role at Actai in Spain. Build production ML systems and neural architectures.

Apply for the Member of Technical Staff, Machine Learning role at Actai in Spain. Build production ML systems and neural architectures.