Jax Jobs in New York City

5 open positions · Updated 4 months ago

Average salary: 355.4k–722.2k/yr

5 roles Senior 4, Mid 1 7 yrs experience Kirkland, Washington, US; Mountain View, California, US; New York City, New York, US +4 more Full-time newest 4 months ago

Also hiring: PyTorch reinforcement learning AI/ML Python async/concurrent programming C++

Showing 5 of 5 positions

Search with filters →

Research Engineer, Machine Learning (RL Velocity) Remote 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 Scientist, Multimodal Alignment, Safety, and Fairness

Google DeepMind Kirkland, Washington, US; Mountain View, California, US; New York City, New York, US $147k–$211k/yr Published 5 months ago

JAX multimodal learning reinforcement learning Python Gemax

Apply →

Performance Engineer, GPU Remote 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, Visual Knowledge Work Remote 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 →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

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

C++ Trio JAX async/concurrent programming Rust

Apply →