"jax" Jobs

37 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 17 of 37 results

Discord

Join Discord as a Senior Machine Learning Engineer to build and deploy models that enhance user safety on the platform.

Discord San Francisco, California, United States $220k–$247.5k/yr Published 2 days ago
Flexible on stack
Goodfire

Join Goodfire as a Machine Learning Engineer to build interpretable AI systems with a world-class team.

Goodfire San Francisco, CA & New York, NY $200k–$400k/yr Published 9 months ago
Postman

Lead research and development of efficient language models at Postman, collaborating with cross-functional teams.

Postman San Francisco, California, United States $218.5k–$288k/yr Published 11 months ago
Flexible on stack
Freenome

Join Freenome as a Senior Machine Learning Engineer to develop and optimize deep learning pipelines for cancer detection.

Freenome Remote $173.8k–$246.8k/yr Published 1 month ago
Flexible on stack
Cantina

Join Cantina as a Research Scientist in Singapore to drive foundational research on video generation models and improve model performance.

Cantina Singapore Published 4 months ago
Flexible on stack
Figma
Figma San Francisco, CA • New York, NY • United States $153k–$376k/yr Published 9 months ago
Preference Model

Join Preference Model as a Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.

Preference Model San Francisco Published 3 days ago
Flexible on stack
Preference Model

Join Preference Model as a Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.

Preference Model San Francisco, United States Published 2 days ago
Flexible on stack
Preference Model

Join Preference Model as a senior ML Engineer to design RL environments for advancing machine learning capabilities.

Preference Model San Francisco Published 2 weeks ago
Flexible on stack
Preference Model

Join Preference Model as a Senior Machine Learning Engineer to design RL environments for advancing ML capabilities.

Preference Model San Francisco, United States Published 2 days ago
Flexible on stack
Anthropic
Anthropic San Francisco, CA $350k–$850k/yr Published 1 year ago
Skild AI

Join Skild AI as a Research Scientist to develop advanced techniques for robotic systems in a passionate and innovative team.

Skild AI Pittsburgh, San Mateo $100k–$300k/yr Published 1 year ago
Flexible on stack