"jax" Jobs
110 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: JAX, PyTorch, Python. Every listing is re-checked daily and closed roles are removed.
Showing 20 of 110 results
Join Krea as an ML Engineer to build personalization and recommendation systems that enhance creative AI tools.
Join SpaceX as an ML Engineer to develop AI surrogate models that enhance engineering simulations for launch vehicles and spacecraft.
Join CoreWeave as an Account Solution Architect to support AI labs and enterprises with GPU infrastructure and MLOps solutions across Northern EMEA.
Join Fireworks AI as a Member of Technical Staff to advance generative AI and multimodal systems through foundational research.
Lead the design and development of next-generation AI agents in a collaborative environment at Postman.
Join Goodfire as a Machine Learning Engineer to build interpretable AI systems with a world-class team.
Lead research and development of efficient language models at Postman, collaborating with cross-functional teams.
Join Mind Robotics as a Research Engineer to build scalable systems for training large ML models in a hands-on robotics environment.
Join Abridge as a Machine Learning Scientist to advance medical NLP and contribute to impactful healthcare solutions.
Join Preference Model as a Research Engineer to advance self-directed learning in machine learning with a focus on RL environments.
Join a dynamic team as an AI Research Engineer to advance AI and machine learning in a startup focused on human-robot interaction.
Develop and implement software solutions for robotic manipulation systems at Skild AI, focusing on state-of-the-art learning-based algorithms.
Join Preference Model as a Research Engineer to advance self-directed learning in large language models within a fast-paced startup.
Join Baseten as a Post-Training Research Engineer to build in-house tooling for efficient and high-quality machine learning models.
Join Freenome as a Senior Machine Learning Engineer to develop and optimize deep learning pipelines for cancer detection.
Related searches