"jax" Jobs

111 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Every listing is re-checked daily and closed roles are removed.

No "jax" jobs in Mountain View right now — showing "jax" jobs in all locations.

Showing 11 of 111 results

Machine Learning Engineer, Reinforcement Learning

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

C++ JAX Robotics reinforcement learning PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on innovative AI hardware solutions.

Graphcore London, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on next-gen AI hardware.

Graphcore Cambridge, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research with a focus on hardware-aware algorithms and impactful implementations.

Graphcore Bristol, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

Research Engineer, Discovery Hybrid Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 1 year ago

Beam AWS Spark JAX AI/ML

Apply →

Research Engineer, Performance RL (Reinforcement Learning) Hybrid Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 5 months ago

Triton JAX CUDA reinforcement learning PyTorch

Apply →

Research Engineer

Join Skild AI as a Research Engineer to develop scalable robotic systems for general-purpose robotic intelligence.

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

JAX imitation learning Machine Learning Robotics reinforcement learning

Flexible on stack

Apply →

Machine Learning Scientist - Open Source Lead

Lead open-source research efforts at Arena Intelligence to advance AI model evaluation and transparency.

Arena Bay Area Published 9 months ago

JAX Machine Learning PyTorch TensorFlow Python

Flexible on stack

Apply →

Staff+ Software Engineer, ML Inference Path Hybrid Visa

Join Anthropic as a Staff Software Engineer to build scalable ML infrastructure for AI safety systems.

Anthropic San Francisco, CA $320k–$485k/yr Published 5 days ago

JAX A/B Testing distributed systems PyTorch TensorFlow

Flexible on stack

Apply →

Research Scientist

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

JAX Machine Learning Robotics reinforcement learning PyTorch

Flexible on stack

Apply →

Staff Machine Learning Scientist Hybrid

Join Freenome as a Staff Machine Learning Scientist to drive innovative AI research in cancer detection.

Freenome Remote $199.7k–$283.5k/yr Published 2 months ago

C++ JAX Java PyTorch TensorFlow

Flexible on stack

Apply →