Direct from source · No middlemen
Senior JAX Jobs
37 open positions · Updated 2 days ago
37 JAX roles across 23 companies, most in AI/ML; 5% fully remote; typical advertised salary $208k
Who is hiring (23 companies)
Role types
Work arrangement: 2 fully remote · 13 hybrid · 8 on-site · 14 not stated
Advertised salaries: p25 $160k · median $208k · p75 $346.2k (from 25 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 37 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 Research Engineer to design and build RL environments for advancing machine learning capabilities.
Join Preference Model as a Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.
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 Instacart as a Senior Machine Learning Engineer to develop ML models for ads response prediction in a fully remote role.
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.