"jax" Jobs
40 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 40 results
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 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 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 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 Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.
Join Preference Model as a new graduate Machine Learning Engineer to design and build reinforcement learning environments.
Join Preference Model as a senior ML Engineer to design RL environments for advancing machine learning capabilities.
Join Preference Model as a Senior Machine Learning Engineer to design RL environments for advancing ML capabilities.
Join Goodfire as the Head of Engineering to lead the development of AI interpretability tools with a world-class team.
Join Anthropic as a Staff Software Engineer to build scalable ML infrastructure for AI safety systems.