JAX Jobs

54 open roles mentioning JAX

Member of Technical Staff, MLE (UK/EU)

10mo ago
Cohere

Cohere

Cohere is seeking a Member of Technical Staff for their Applied ML team. In this role, you will collaborate directly with customers to understand their challenges and implement solutions leveraging Large Language Models. You will apply your problem-solving skills, creativity, and technical expertise to bridge the gap in enterprise AI adoption, delivering impactful products and disrupting key industries. This is an opportunity to join at a pivotal moment, shape the company's offerings, and contribute to cutting-edge AI development.

London hybrid FullTime
CoherePythonTensorFlow +2 more

AI Scientist - Audio

10mo ago
Mistral AI

Mistral AI

Mistral is on a mission to democratize AI by producing frontier intelligence for everyone, developed in the open. We are a dynamic, collaborative team passionate about AI's potential to transform society, with diverse teams distributed globally. We develop models for enterprise and consumers, focusing on systems that change business operations and integrate into daily lives, while also releasing frontier models open-source. We are hiring experts in large language model training and distributed systems to shape the future of AI.

Paris remote Full-time
MistralKubernetesPython +8 more

Member of Technical Staff, Integration/RL Team (Research Engineer)

11mo ago
Cohere

Cohere

Cohere is a leading enterprise AI company focused on building cutting-edge foundation AI models and end-to-end products for real-world business problems. The integration team specifically focuses on developing and scaling machine learning algorithms and infrastructure for LLM post-training, with an emphasis on large-scale, distributed Reinforcement Learning (RL) methods. This role is crucial for enhancing the post-training codebase by implementing new research tools, optimizing algorithms, and scaling distributed RL capabilities. We are looking for passionate individuals who are meticulous in their approach to engineering and science, contributing to both production code and research efforts.

Paris remote FullTime
CohereKubernetesPython +3 more

Member of Technical Staff, Post-Training

1y ago
Cohere

Cohere

Cohere is seeking a Member of Technical Staff to focus on post-training of AI models. This role is crucial for advancing the state of the art in model post-training and shipping cutting-edge models to production, bridging the gap between research and practical application. You will have access to significant compute resources and a talented team to contribute to increasing model capabilities and driving customer value. The position offers a unique opportunity to contribute to both production code and research efforts, depending on your interests and organizational needs.

London hybrid FullTime
CohereKubernetesPython +3 more

Member of Technical Staff, Search

1y ago
Cohere

Cohere

Cohere is seeking talented individuals to join our Search team as a Member of Technical Staff. This role focuses on developing state-of-the-art models for information retrieval, including training embedding and reranker models. You will have the opportunity to revolutionize search experiences by building intelligent, efficient, and precise search systems. Your work will advance semantic search techniques, improve accuracy and efficiency, and involve integrating novel technologies into our search infrastructure. We are looking for someone passionate about search with a strong background in information retrieval and experience working with diverse technologies and cross-functional teams.

United States remote FullTime
CoherePythonGo +5 more

Applied Scientist / Research Engineer - Singapore

1y ago
Mistral AI

Mistral AI

Mistral AI is seeking Applied Scientists and Research Engineers to drive innovative research and collaborate with clients on complex research projects. You will develop state-of-the-art models across different modalities such as text, image, and speech. By developing novel methods and research ideas, you will apply these models across a diverse set of use cases and domains. Working cross-functionally with both external and internal science, engineering, and product teams, you will deliver high-impact AI solutions that make a significant difference. We are a dynamic, collaborative team passionate about AI and its potential to transform society, with a diverse workforce thriving in competitive environments and committed to driving innovation.

Singapore onsite Full-time
MistralPythonRAG +8 more

Applied Scientist / Research Engineer (Internship)

1y ago
Mistral AI

Mistral AI

Mistral AI is seeking Applied Scientists Interns and Research Engineers Interns to drive innovative research and collaborate with clients on complex research projects. You will develop state-of-the-art models across different modalities such as text, image, and speech. By developing novel methods and research ideas, you will apply these models across a diverse set of use cases and domains. Working cross-functionally with both external and internal science, engineering, and product teams, you will deliver high-impact AI solutions that turn the needle. This position is open for our local offices in Paris and London.

Paris onsite Full-time
MistralPythonRAG +7 more

Senior/Staff Applied Scientist/Research Engineer, EMEA

1y ago
Mistral AI

Mistral AI

Mistral AI is seeking Applied Scientists and Research Engineers to drive innovative research and collaborate with clients on complex research projects. You will develop state-of-the-art models across different modalities such as text, image, and speech. By developing novel methods and research ideas, you will apply these models across a diverse set of use cases and domains. Working cross-functionally with both external and internal science, engineering, and product teams, you will deliver high-impact AI solutions that make a significant difference. We are a dynamic, collaborative team passionate about AI and its potential to transform society. Our diverse workforce thrives in competitive environments and is committed to driving innovation. Join us to be part of a pioneering company shaping the future of AI.

Paris remote Full-time
MistralPythonRAG +8 more

Senior Member of Technical Staff, Multimodal AI

1y ago
Cohere

Cohere

Cohere is seeking a Senior Member of Technical Staff focused on Multimodal AI to join their cutting-edge enterprise AI company. This role involves designing and developing advanced multimodal AI systems that integrate text, speech, and vision, pushing the boundaries of what's possible in AI. You will have access to exceptional compute resources and collaborate with world-class teams to innovate and shape the future of AI. The position is ideal for individuals passionate about machine learning and its real-world applications, who enjoy optimizing large models and thrive in a fast-paced, technically challenging environment.

San Francisco remote FullTime
OpenAIMistralCohere +5 more

Member of Technical Staff, MLE (Korea)

1y ago
Cohere

Cohere

Cohere is seeking a Member of Technical Staff for their Applied ML team. In this role, you will collaborate directly with customers to understand their challenges and implement solutions leveraging Large Language Models. You will apply your problem-solving skills, creativity, and technical expertise to bridge the gap in Enterprise AI adoption, delivering impactful products and disrupting key industries. This is an opportunity to join at a pivotal moment, shape the company's offerings, and contribute across multiple facets of the business.

Korea remote FullTime
CoherePythonTensorFlow +2 more

Open-Source Software, Machine Learning Engineer

1y ago
Mistral AI

Mistral AI

Mistral AI is democratizing AI through high-performance, optimized, open-source models, products, and solutions. We are a dynamic, collaborative team passionate about AI's potential to transform society, with teams distributed globally. We are seeking an Open-Source Software, Machine Learning Engineer to join our OSS team, which is embedded within our Science team. This role is critical in helping turn research breakthroughs into tangible solutions and improving Mistral's open-source ecosystem by open-sourcing state-of-the-art models and maintaining our publicly available libraries.

Paris remote Full-time
MistralPythonGo +10 more

Research Engineer, Machine Learning - Paris/London/Zurich/Warsaw

1y ago
Mistral AI

Mistral AI

Mistral AI is a pioneering company focused on democratizing AI through high-performance, optimized, open-source models and solutions. We aim to simplify tasks, save time, and enhance learning and creativity by integrating AI seamlessly into daily working life. Our comprehensive AI platform serves both enterprise and personal needs, featuring offerings like Le Chat, La Plateforme, Mistral Code, and Mistral Compute. We are a dynamic, collaborative, and diverse team passionate about AI's potential to transform society, driven by innovation and a low-ego, team-spirited culture.

Paris hybrid Full-time
MistralPythonPyTorch +11 more

AI Scientist - Palo Alto

2y ago
Mistral AI

Mistral AI

Mistral is on a mission to democratize AI by producing frontier intelligence for everyone, developed in the open. We are a dynamic, collaborative team passionate about AI and its potential to transform society, with diverse teams distributed across Europe, the USA, and Asia. We develop models for enterprise and consumers, focusing on systems that change business operations and integrate into daily lives, while also releasing frontier models open-source. We are hiring experts in the training of large language models and distributed systems to join our pioneering company shaping the future of AI.

Palo Alto onsite Full-time
MistralKubernetesPython +8 more

AI Scientist - Paris/London - Onsite or Hybrid or Remote

3y ago
Mistral AI

Mistral AI

Mistral is on a mission to democratize AI by producing frontier intelligence for everyone, developed in the open. We are a dynamic, collaborative team passionate about AI's potential to transform society, with diverse teams distributed globally. We develop models for enterprise and consumers, focusing on systems that change business operations and integrate into daily lives, while also releasing frontier models open-source. We are hiring experts in the training of large language models and distributed systems to join our pioneering company shaping the future of AI.

Paris remote Full-time
MistralKubernetesPython +8 more

Get new AI jobs in your inbox

A weekly digest of the newest AI engineering roles.

© 2026 AI Job Board. All rights reserved.