Direct from source · No middlemen
JAX Jobs
122 open positions · Updated 15 hours ago
122 JAX roles across 52 companies, most in AI/ML; 5% fully remote; typical advertised salary $200k
Who is hiring (52 companies)
Work arrangement: 6 fully remote · 31 hybrid · 30 on-site · 55 not stated
Advertised salaries: p25 $151.5k · median $200k · p75 $306.5k (from 47 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 122 positions
Search with filters →Join Atoms as an AI Research Intern to bridge AI research and real-world applications in autonomous transport platforms.
Join Skydio as an intern to develop reinforcement learning methods for autonomous drones in real-world applications.
Join Eka Robotics as a Machine Learning Engineer to develop cutting-edge algorithms for robotic manipulation in a fast-paced startup environment.
Join Eka Robotics as a Machine Learning Engineer to develop cutting-edge computer vision systems for robotic manipulation.
Join Eka Robotics as a Machine Learning Infrastructure Engineer to build and maintain large-scale model training systems for robotics.
Own and improve the development experience by designing and maintaining build systems and developer tooling at Eka Robotics.
Join K2 Space as a Photonic IC Design Engineer to lead photonic design automation and modeling for cutting-edge satellite technology.
Join Graphcore as a Software Engineer to enhance AI frameworks and optimize performance for machine learning workloads.
Join Reflection AI as a Member of Technical Staff to design and implement data sourcing and curation techniques for AI models.
Join Atoms as an AI/ML Research Intern to bridge high-level AI research with real-world physical actuation for autonomous transport platforms.
Lead data science initiatives to enhance user engagement and ecosystem understanding at Reddit.
Join Preference Model as a Research Engineer to advance self-directed learning in large language models within a fast-paced startup.
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 Graphcore as a Graduate Machine Learning Engineer and gain hands-on experience in AI systems over a two-year program.
Join Graphcore as a Graduate Software Engineer to work on AI/ML Test Systems with hands-on learning and development opportunities.
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.