"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

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
Skild AI

Join Skild AI as a Robotics Engineer to develop software solutions for robotic manipulation systems using state-of-the-art algorithms.

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

Join Arena as a Machine Learning Scientist to evaluate AI models and contribute to impactful research in a collaborative environment.

Arena Bay Area Published 8 months ago
Flexible on stack 70% coding
Twilio

Join Twilio as a Machine Learning Engineer to design AI-powered features that enhance customer conversations.

Twilio Remote - Ireland Published 2 months ago
Flexible on stack
Skild AI

Join Skild AI as a Research Engineer to enhance and deploy robotic intelligence in real-world environments.

Skild AI San Mateo, CA $100k–$300k/yr Published 2 years ago
Flexible on stack
Skild AI

Join Skild AI as a Software Engineer to develop and optimize software infrastructure for training advanced AI models in robotics.

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

Develop and refine deep learning models for robotic manipulation at a pioneering AI startup.

Skild AI Pittsburgh, San Mateo $100k–$300k/yr Published 1 year ago
Flexible on stack
Figma
Figma San Francisco, CA • New York, NY • United States $153k–$376k/yr Published 8 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 2 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 1 day ago
Flexible on stack
Preference Model

Join Preference Model as a new graduate Machine Learning Engineer to design and build reinforcement learning environments.

Preference Model San Francisco Published 4 months 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 1 day ago
Flexible on stack
Skild AI

Join Skild AI as a Machine Learning Engineer to develop cutting-edge reinforcement learning algorithms for robotic applications.

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

Join Goodfire as the Head of Engineering to lead the development of AI interpretability tools with a world-class team.

Goodfire San Francisco, CA $300k–$450k/yr Published 8 months ago
Flexible on stack
Skild AI

Join Skild AI as a Machine Learning Engineer to develop cutting-edge reinforcement learning algorithms for adaptable robotic systems.

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