Jax Jobs in New York

9 open positions · Updated 1 week ago

Average salary (USD/year): 312.9k–604.5k/yr

9 roles Senior 6, Staff 2, Mid 1 5–7 yrs experience New York City (Remote possible for exceptional candidates) +8 more Full-time newest 1 week ago

Also hiring: PyTorch Python AI/ML C++ Kubernetes Machine Learning

Showing 9 of 9 positions

Search with filters →

Machine Learning Engineer Hybrid

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

JAX distributed systems PyTorch Python

Apply →

Member of Technical Staff - Research Software Engineer Visa

Join Reflection AI as a Research Software Engineer to bridge research and production in cutting-edge AI training systems.

Reflection AI New York, NY Published 6 months ago

Ray JAX Slurm RDMA PyTorch

Flexible on stack

Apply →

Member of Technical Staff - Quantitative Research

Join Udio as a full-stack scientist to lead quantitative research efforts at the intersection of music and AI.

Udio New York City (Remote possible for exceptional candidates) $250k–$350k/yr Published 7 months ago

JAX Machine Learning Apache Beam TensorFlow GCP

Flexible on stack

Apply →

Senior Research Engineer

Join AssemblyAI as a Senior Research Engineer to enhance large-scale distributed training and inference systems in Voice AI.

AssemblyAI Remote - New York $270k–$310k/yr Published 1 week ago

C++ JAX Rust TPU Python

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 →

AI Applied Scientist Hybrid

Figma San Francisco, CA • New York, NY • United States $153k–$376k/yr Published 8 months ago

C++ JAX Machine Learning AI/ML Java

Apply →

Performance Engineer, GPU Hybrid Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 11 months ago

FP8 NVLink Triton JAX CUTLASS

Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Hybrid Visa

Anthropic San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 year ago

C++ Trio JAX async/concurrent programming Rust

Apply →

Research Engineer, Visual Knowledge Work Hybrid Visa

Anthropic New York City, NY; San Francisco, CA; Seattle, WA $350k–$850k/yr Published 7 months ago

JAX Large Language Models Machine Learning AI/ML reinforcement learning

Apply →