Jax In Jobs

110 open positions · Updated 1 week ago

Average salary (USD/year): 300k–638.3k/yr

Showing 10 of 110 positions

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 →

Research Engineer, Machine Learning (RL Velocity) Hybrid Visa

Anthropic Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY $500k–$850k/yr Published 4 months ago

JAX AI/ML PyTorch

Apply →

Research Engineer, Machine Learning (RL Velocity) Hybrid Visa

Anthropic London, UK £370k–£630k/yr Published 4 months ago

JAX AI/ML PyTorch

Apply →

Research Engineer, Pretraining Scaling - London Visa

Anthropic London, UK £260k–£630k/yr Published 11 months ago

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Apply →

Software Engineer, AI Training and Infrastructure

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

C++ AWS JAX Google Cloud cloud-based training environments

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 4 days ago

JAX A/B Testing distributed systems PyTorch TensorFlow

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 8 months ago

JAX Machine Learning PyTorch TensorFlow Python

Flexible on stack

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, Pretraining Scaling Visa

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

ML frameworks JAX model operations AI/ML large-scale distributed systems

Apply →

Research Engineer, Discovery Hybrid Visa

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

Beam AWS Spark JAX AI/ML

Apply →

← Prev Page 6 of 6