← Back to jobs

JAX

116 jobs tagged JAX

Median salary: $199,675 (from 55 disclosed salaries, annual, USD) 5% remote Top hiring: Anthropic, Skild AI, Graphcore, DeepL, Perplexity

JAX jobs — quick facts

How many JAX jobs are available?
116 active JAX roles sourced directly from company career pages, updated daily on Joblaze.
What is the median salary for JAX jobs?
The median advertised salary for JAX roles on Joblaze is about $199,675 per year.
Which companies are hiring for JAX?
Companies currently hiring JAX include Anthropic, Skild AI, Graphcore, DeepL, Perplexity.
Are JAX jobs remote?
5% of active JAX roles on Joblaze offer remote work.
Inferact

Join Inferact as a co-op student to work on cutting-edge AI inference systems in a hands-on engineering role.

Inferact San Francisco, California, United States Published 1 day ago
Flexible on stack
Anthropic

Join Anthropic as a Staff Software Engineer to build scalable ML infrastructure for AI safety systems.

Anthropic San Francisco, CA $320k–$485k/yr Published 2 days ago
Flexible on stack
Fireworks AI

Join Fireworks AI as a Member of Technical Staff to advance generative AI and multimodal systems through foundational research.

Fireworks AI San Mateo Published 3 days ago
Flexible on stack
Cantina

Join Cantina as a Research Intern to work on next-gen video models in a hands-on, mentorship-driven environment.

Cantina Singapore Published 4 days 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 1 week ago
Flexible on stack
DeepL

Lead scientific innovation in speech and translation models for real-time voice products at DeepL.

DeepL London Published 4 weeks ago
Flexible on stack 70% coding
DeepL

Lead research on fine-tuning and steerability of LLM-based translation models in a collaborative AI-focused environment.

DeepL London Published 1 month ago
Flexible on stack
DeepL

Join DeepL as a Senior Research Scientist to innovate in reinforcement learning and shape the future of AI technology.

DeepL London Published 1 month ago
Flexible on stack
DeepL

Join DeepL as a Senior Research Scientist to drive foundational modeling decisions for next-gen translation systems.

DeepL London Published 1 month ago
Flexible on stack
DeepL

Lead fine-tuning and development of multimodal models for document translation at DeepL.

DeepL London Published 1 month ago
Flexible on stack
ChipAgents

Join ChipAgents as a Research Scientist to advance AI-assisted electronic design automation in a high-impact, mission-driven team.

ChipAgents San Jose $150k–$350k/yr Published 3 months ago
Flexible on stack
Xaira Therapeutics

Join Xaira Therapeutics as a Biomedical AI Engineer to design AI systems that accelerate therapeutic discovery.

Xaira Therapeutics South San Francisco, California, United States $227k–$295k/yr Published 5 months ago
Flexible on stack
Xaira Therapeutics

Join Xaira Therapeutics as an AI Scientist to develop multimodal foundation models for drug discovery using advanced AI techniques.

Xaira Therapeutics South San Francisco, California, United States $170k–$240k/yr Published 4 months ago
Flexible on stack
Xaira Therapeutics

Join Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.

Xaira Therapeutics Seattle, Washington, United States $189k–$245k/yr Published 3 months ago
Flexible on stack
Goodfire

Join Goodfire as a Forward Deployed Engineer to integrate AI interpretability solutions with strategic partners.

Goodfire San Francisco, CA $200k–$325k/yr Published 5 months ago
Flexible on stack 70% coding
Goodfire

Join Goodfire as a Machine Learning Engineer to build interpretable AI systems with a world-class team.

Goodfire San Francisco, CA & New York, NY $200k–$400k/yr Published 9 months ago
Goodfire

Join Goodfire as the Head of Engineering to lead the development of AI interpretability tools with a world-class team.

Goodfire San Francisco, CA $300k–$450k/yr Published 7 months ago
Flexible on stack
CuspAI

Join CuspAI as an Applied AI/ML Engineer to design intelligent agents for autonomous materials discovery.

CuspAI Amsterdam, NL Published 4 weeks ago
Flexible on stack
Skild AI

Join Skild AI as a Computer Vision AI & ML Engineer to develop advanced perception systems for robotics and automation.

Skild AI Bengaluru, India Published 9 months ago
Flexible on stack
Skild AI

Join Skild AI as a Research Engineer to enhance and deploy robotic intelligence in real-world environments.

Skild AI San Mateo, CA $100k–$300k/yr Published 2 years ago
Flexible on stack
Skild AI

Join Skild AI as a Research Scientist to develop advanced techniques for robotic systems in a passionate and innovative team.

Skild AI Pittsburgh, San Mateo $100k–$300k/yr Published 1 year ago
Flexible on stack
Skild AI

Join Skild AI as a Machine Learning Engineer to develop cutting-edge reinforcement learning algorithms for robotic applications.

Skild AI San Mateo, Pittsburgh $100k–$300k/yr Published 1 year ago
Flexible on stack
Skild AI

Join Skild AI as a Software Engineer to optimize AI inference for robotic systems, enhancing their performance and adaptability.

Skild AI San Mateo, CA $100k–$300k/yr Published 1 year ago
Flexible on stack
Skild AI

Join Skild AI as a Robotics Engineer to develop software solutions for robotic manipulation systems using state-of-the-art algorithms.

Skild AI San Mateo, CA $100k–$300k/yr Published 1 year ago
Flexible on stack
Skild AI

Develop and implement software solutions for robotic manipulation systems at Skild AI, focusing on state-of-the-art learning-based algorithms.

Skild AI San Mateo $100k–$300k/yr Published 11 months ago
Flexible on stack
Skild AI

Join Skild AI as a Research Engineer to develop scalable robotic systems for general-purpose robotic intelligence.

Skild AI Pittsburgh, San Mateo $100k–$300k/yr Published 1 year ago
Flexible on stack
Skild AI

Join Skild AI as a Computer Vision AI & ML Engineer to develop advanced perception systems for robotics and automation.

Skild AI San Mateo, CA Published 3 months ago
Flexible on stack
Skild AI

Join Skild AI as a Software Engineer to develop and optimize software infrastructure for training advanced AI models in robotics.

Skild AI San Mateo, CA $100k–$300k/yr Published 1 year ago
Flexible on stack
Skild AI

Join Skild AI as a Machine Learning Engineer to develop cutting-edge reinforcement learning algorithms for adaptable robotic systems.

Skild AI Pittsburgh, San Mateo $100k–$300k/yr Published 2 years ago
Flexible on stack
Skild AI

Develop and refine deep learning models for robotic manipulation at a pioneering AI startup.

Skild AI Pittsburgh, San Mateo $100k–$300k/yr Published 1 year ago
Flexible on stack
Proxima Fusion

Join Proxima Fusion as a Computational Engineer to tackle complex engineering challenges in the development of commercial fusion power plants.

Proxima Fusion Munich Published 2 months ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.

Perplexity AI London Published 5 months ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Technical Staff member to enhance our AI inference engine with cutting-edge technologies.

Perplexity AI San Francisco Published 5 months ago
Flexible on stack
Perplexity AI

Join Perplexity AI as an AI Inference Engineer to optimize and develop our inference engine for various model architectures.

Perplexity AI London Published 5 months ago
Flexible on stack
Perplexity AI

Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.

Perplexity AI Belgrade Published 8 months ago
Flexible on stack
Reflection AI

Join Reflection AI as a Research Software Engineer to bridge research and production in cutting-edge AI training systems.

Reflection AI New York, NY Published 6 months ago
Flexible on stack
Postman

Lead research and development of efficient language models at Postman, collaborating with cross-functional teams.

Postman San Francisco, California, United States $218.5k–$288k/yr Published 11 months ago
Flexible on stack
Postman

Lead the design and development of next-generation AI agents in a collaborative environment at Postman.

Postman San Francisco, California, United States $256k–$276k/yr Published 11 months ago
Flexible on stack
Armada

Join Armada as an AI Engineer to build and deploy intelligent systems at the edge of the physical world.

Armada Bellevue Office, Sunset Corporate Campus $154.6k–$193.2k/yr Published 1 month ago
Flexible on stack 70% coding
baseten

Join Baseten as a Post-Training Research Engineer to build in-house tooling for efficient and high-quality machine learning models.

baseten San Francisco Published 5 months ago
baseten

Join Baseten as a Forward Deployed Engineer to solve complex AI challenges for leading companies.

baseten San Francisco Published 3 weeks ago
Flexible on stack
Netic

Join Netic as an Applied AI Research Engineer to drive impactful ML projects in a cutting-edge AI-first environment.

Netic San Francisco Published 1 year ago
Flexible on stack
Graphcore

Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware in a dynamic environment.

Graphcore Cambridge, UK Published 9 months ago
Flexible on stack
Graphcore

Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware in a collaborative environment.

Graphcore Bristol, UK Published 9 months ago
Flexible on stack
Graphcore

Join Graphcore's Triton team to enhance AI frameworks and optimize software for cutting-edge machine learning workloads.

Graphcore Gdańsk, Pomeranian Voivodeship, Poland Published 5 months ago
Flexible on stack
Graphcore

Join Graphcore's Triton team to enhance AI frameworks and improve performance on cutting-edge hardware.

Graphcore Bristol, UK; Gdańsk, Pomeranian Voivodeship, Poland Published 2 weeks ago
Flexible on stack
Graphcore

Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware.

Graphcore London, UK Published 9 months ago
Flexible on stack
Graphcore

Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware.

Graphcore Gdańsk, Pomeranian Voivodeship, Poland PLN 260.4k–PLN 352.2k/yr Published 2 months ago
Flexible on stack
Graphcore

Join Graphcore as an AI Research Engineer to advance AI research with a focus on hardware-aware algorithms and impactful implementations.

Graphcore Bristol, UK Published 1 month ago
Flexible on stack
Graphcore

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on next-gen AI hardware.

Graphcore Cambridge, UK Published 1 month ago
Flexible on stack