Direct from source · No middlemen
Senior JAX Jobs
35 open positions · Updated 4 days ago
35 JAX roles across 22 companies, most in AI/ML; typical advertised salary $220k
Who is hiring (22 companies)
Role types
Work arrangement: 13 hybrid · 8 on-site · 14 not stated
Advertised salaries: p25 $171k · median $220k · p75 $347.0k (from 23 disclosed annual salaries, USD)
Counts are open roles Joblaze currently tracks on company career pages for this exact skill/location; salaries are advertised minimums, annual, converted to USD.
Showing 20 of 35 positions
Search with filters →Join Eka Robotics as a Machine Learning Engineer to develop cutting-edge computer vision systems for robotic manipulation.
Join K2 Space as a Photonic IC Design Engineer to lead photonic design automation and modeling for cutting-edge satellite technology.
Join Preference Model as a Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.
Join Preference Model as a Research Engineer to design and build RL environments for advancing machine learning capabilities.
Join Mecka AI as a Research Scientist to advance multimodal foundation models and tackle challenging research problems.
Join Graphcore as a Senior Machine Learning Engineer to advance AI models on cutting-edge hardware in a hybrid work environment.
Join Graphcore as a Senior Machine Learning Engineer to advance AI models on cutting-edge hardware in a hybrid work environment.
Join Graphcore as a Senior Machine Learning Engineer to advance AI models on cutting-edge hardware in a hybrid work environment.
Join Anthropic as a Research Engineer/Scientist to develop next-generation RL algorithms and architectures at frontier scale.
Join Discord as a Senior Machine Learning Engineer to build and deploy models that enhance user safety on the platform.
Join Periodic Labs as a Computational Scientist to develop differentiable simulations for continuum-physics problems using AI.
Join ChipAgents as a Research Scientist to advance AI-assisted electronic design automation in a high-impact, mission-driven team.
Join Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.
Join Goodfire as a Machine Learning Engineer to build interpretable AI systems with a world-class team.
Join Skild AI as a Research Scientist to develop advanced techniques for robotic systems in a passionate and innovative team.
Join Netic as an Applied AI Research Engineer to drive impactful ML projects in a cutting-edge AI-first environment.
Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware.
Develop and apply methods at the intersection of 3D molecular simulation and machine learning to impact drug discovery programs.