JAX jobs
32 roles posted Oct 20–26, 2026 list JAX as a skill, from 23 employers. That is 1% of the 5,587 developer roles RoleWeek lists.
A role counts when JAX is among the up to five technologies RoleWeek read in its posting. Data as of .
Who is hiring
| Employer | Roles |
|---|---|
| Embedding VC | 3 |
| SandboxAQ | 3 |
| Waymo | 3 |
| Amazon | 2 |
| NVIDIA | 2 |
| Oracle | 2 |
| Atoms | 1 |
| CMU | 1 |
| Electronic Arts | 1 |
| Expedia Group | 1 |
The 10 employers with the most roles, out of 23. One role is one opening, however many places it is listed for.
By level
| Level | Roles | Share |
|---|---|---|
| Intern | 6 | 19% |
| Entry level | 2 | 6% |
| Mid level | 10 | 31% |
| Senior | 9 | 28% |
| Staff+ | 5 | 16% |
Level is RoleWeek’s reading of each posting.
What these roles post as salary
17 of these 32 roles list a yearly salary range in US dollars. The median bottom of those ranges is $134K and the median top is $249K; the middle half of the bottoms runs from $124K to $152K, and of the tops from $215K to $252K. These are the ranges employers posted in their own postings: not offers, not what people are paid and not a market average. Posted ranges across all the developer roles RoleWeek lists: Developer salary ranges by city.
Not counted: 10 roles list no salary and 5 list pay by the hour. Pay by the hour, week or month is never converted or mixed with yearly ranges, and each page uses one currency.
Where the roles are
| City | Roles |
|---|---|
| San Francisco Bay Area | 16 |
| Remote | 6 |
| New York | 4 |
| Elsewhere in the US & Canada | 4 |
| Washington DC | 2 |
| Seattle | 1 |
| Boston | 1 |
A role listed for several cities counts in each. Only cities with 15 or more roles have a page of their own.
Roles that list JAX
Showing 24 of 32, newest first. Search the roles in this list that mention JAX
Autonomy Engineer Intern, Reinforcement Learning, 2027 (HQ)
SkydioSan Mateo, CAOn-site$47/hrInternshipJAXPythonBacked by a16zMachine Learning Scientist – Generative Molecular Design
SandboxAQUSRemote$134.4K–$252K/yrJAXLLMs4+ yrsMachine Learning Scientist, Generative Molecular Design (Material Discovery)
SandboxAQUSRemote$134.4K–$252K/yrJAXLLMs4+ yrsSenior AI Reverse Engineering Researcher
CMULinthicumJAXXGBoost2027 Summer Intern, MS/PhD, Software Engineer, ML Frameworks and Efficiency
WaymoMountain View, CA$70–$85/hrInternshipJAXPyTorchBacked by a16zArtificial Intelligence (AI) Subject Matter Expert Support (SME) (Part-Time)
GDITFort Belvoir, VAPart-timeJAXLLMs8+ yrsMember of Technical Staff, Robotics Research
SieveSan FranciscoOn-siteJAXSwift4+ yrsPrincipal ML Engineer
Electronic ArtsRedwood CityHybridJAXPyTorch20+ yrsAI & I Data Analytics Research Fellow – Radiation Oncology
Mayo ClinicRochester, MN$72K–$84.2K/yrJAXPythonAI Engineer IV
PremeraTelecommuterRemote$131.9K–$270.5K/yrJAXTensorFlowSenior2027 Future Talent Program - Data, AI, and Genome Sciences, Foundation Models - Intern
MerckCambridge, MA$39.9K–$111.1K/yrJAXPythonMember of Technical Staff, AI Engineering for Biosecurity
Radical NumericsSan FranciscoJAXPythonPrincipal Applied Scientist
OracleUS$126.2K–$264.1K/yrJAXLLMsNo account needed6+ yrsRobotics Engineering Intern — Hardware & Software
TraceNew YorkOn-site$20–$50/hrInternshipJAXComputer VisionSenior Applied Scientist
OracleUS$113K–$234.6K/yrJAXLLMsNo account needed3+ yrsSenior Applied Scientist – AI/ML Calibration & Optimization
KLAMilpitas, CA$136.3K–$231.7K/yrJAXPython2027 Summer Intern, PhD, AI Foundations - Recipes
WaymoMountain View, CA$85/hrInternshipJAXPythonBacked by a16zMachine Learning Engineer, Scenes
WaymoMountain View, San Francisco$175K–$215K/yrJAXPythonBacked by a16zSenior Robotics/AI Motor Control Scientist, Fauna
AmazonNew York$183.8K–$248.7K/yrJAXJava6+ yrsML/AI Research Internship - PhD
AtomsSan Francisco, CA$70/hrInternshipJAXPyTorchNo account neededMember of Technical Staff - Pre-Training Data
ReflectionSan Francisco, CAOn-siteJAXPythonBacked by SequoiaMember of Technical Staff — Agent Post-Training
Embedding VCSan Francisco Bay AreaOn-siteJAXPythonMember of Technical Staff — Post-Training Infrastructure
Embedding VCSan Francisco Bay AreaOn-siteJAXPython
How these counts are made
- Source: the roles RoleWeek lists as posted in the last 7 days, straight from employers’ own hiring systems. A role counts for a skill when that skill is one of the up to five technologies RoleWeek read in the posting’s title and description (the title first, then the most mentioned). A role that names a skill once among many tools may not list it, so these count the roles that list the skill, not all the roles that touch it.
- Names are matched exactly: Java is not JavaScript, C++ and C# are not C and not each other, Go is the language (never “go-to-market”) and R is the language. A posting that names a vendor only as its own employer, an investor or a customer does not count for that vendor’s product.
- One role is one card: an opening listed for several places counts once on a page. A role listed in several cities counts on each city’s page.
- A skill has a page when 15 or more roles list it, and a city page when 15 or more do there and it is not the same set of roles as the skill’s own page. Practices that are not one technology (CI/CD, Git, microservices, networking) have none.
- Salaries are only what employers posted, summarised by the rules of the salary pages: yearly ranges only, one currency per page, a median shown with 15 or more roles. The numbers are rebuilt every day.
The full list: All developer roles (5,587 roles posted in the last 7 days). Every skill: developer roles by skill.