JAX jobs
JAX: Google’s library for fast numerical computing and AI research. Waymo, The Jackson Laboratory's, xAI, Anthropic and Cohere have the most open roles mentioning JAX right now.
RSS feedMy jobs then shows how many are new each time you come back. No account, no email.
375 open roles · 18 posted in the last 7 days · 10% remote · 111 companies hiring · $238K median advertised salary · checked Nov 15 against employer career pages
Job titles in JAX roles
Bars show the share of these 375 roles; one role can count in several rows.
Newest JAX jobs
- Sales Instructional DesignerThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut; Remote$70K - $101KRemoteJAXInstructional designLMS
- Director, Public Relations & CommunicationsThe Jackson Laboratory's · 🇺🇸New York, New York; Farmington, Connecticut; Remote$146K - $212KRemoteLead and aboveJAX
- Staff/Sr. Staff/Principal AI research scientistWriter · 🇺🇸San Francisco, CA$234K - $384KHybridLead and abovePythonMachine LearningLLM
- AI research scientistWriter · 🇺🇸San Francisco, CA$168K - $233KHybridPythonMachine LearningLLM
- Cellular Biologist IIThe Jackson Laboratory's · 🇺🇸New York, New York—JAXInventory managementProject management
- Postdoctoral AssociateThe Jackson Laboratory's · 🇺🇸New York, New York · +1 more location—JAXTest equipmentStatistics
- Packaging Thermal and Mechanical Analysis Internship - Summer 2027Lightmatter · 🇺🇸Mountain View, CA—InternPythonMachine LearningPyTorch
- Member of Technical Staff - ML Infrastructure Engineer, RL Post-trainingPreference Model · 🇺🇸San Francisco$200K - $350KGCPLLMPyTorch
- Member of Technical Staff - Research Engineer, Post-trainingPreference Model · 🇺🇸San Francisco$200K - $350KPythonMachine LearningLLM
- Member of Technical Staff - Research Engineer, RL EnvironmentsPreference Model · 🇺🇸San Francisco$200K - $350KPythonMachine LearningLLM
- Controls Automation Engineer - MaintenanceThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut$92K - $134KPythonC++Azure
- Research Engineering Manager, STEMTuring · 🇺🇸United States$260K - $350KPythonPostgreSQLDocker
- Field Service Mechanical Technician - JAXIndustrial Electric Manufacturing · 🇺🇸Jacksonville, Florida, United States—JAXPowerPointMicrosoft Word
- Sr. Model Engineer, RL DataCursor · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Model Engineer, RL DataCursor · 🇺🇸Palo Alto, CA$160K - $250KRustPythonC++
- AI Engineer - PCoupa · 🇮🇳Pune, India—PythonGCPLLM
- AI Engineer - 11967Coupa · 🇮🇳Pune, India—PythonGCPLLM
- Member of Technical Staff, Pretraining DataMirendil · 🇺🇸San Francisco$300K - $400KPythonPyTorchLLM evaluation
- Sr. Model Engineer, VoicexAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- GPU Kernel EngineerSciforium · 🇺🇸San Francisco, CA$190K - $250KPythonC++LLM
- Senior Forward Deployed EngineerSuki · 🇺🇸Redwood City, CA$225K - $250KSeniorGoPythonTypeScript
- 2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable SimulationWaymo · 🇺🇸Mountain View, California, United States—InternPythonMachine LearningPyTorch
- 2027 Summer Intern, Machine Learning, MS/PhD, Simulation Realism (World Model)Waymo · 🇺🇸Mountain View, California, United States—InternPythonMachine LearningPyTorch
- Applied ResearcherRiver AI Inc. · 🇺🇸Palo Alto, CA$200K - $420KMachine LearningPyTorchReinforcement learning
- Staff Machine Learning Engineer, AI InfrastructureWaymo · 🇺🇸Mountain View, CA, USA; San Francisco, CA, USA; New York…$251K - $310KLead and abovePythonC++Machine Learning
- Research Scientist, Discovery (Engineering & Applied Sciences)Anthropic · 🇺🇸San Francisco, CA$500K - $850KPythonMachine LearningLLM
- Head of Applied ResearchKrakentech · 🇬🇧London, UK—HybridLead and aboveMachine LearningLLMPyTorch
- Senior Study Director, Rare Disease Translational CenterThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine$99K - $166KLead and aboveRJAXData quality
- Sr. Model Engineer, RL Training FrameworkxAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Senior Research EngineerFar.ai · Berkeley Office$225K - $350KHybridSeniorPython.NETMachine Learning
- Senior Software EngineerLancedb · Americas or APAC$180K - $250KSeniorRustPythonJava
- Senior Machine Learning Engineer (Growth)Discord · 🇺🇸San Francisco Bay Area$220K - $275KSeniorPythonMachine LearningPyTorch
- Senior Staff Machine Learning Engineer, Perception, Simulation RealismWaymo · 🇺🇸Mountain View, California, United States; San Francisco…$298K - $368KLead and abovePythonC++Machine Learning
- Quantum Characterization, Verification, and Validation EngineerPsiQuantum · 🇺🇸Milpitas, California$89K - $149KPythonPyTorchTensorFlow
- Machine Learning Engineer - New GradSambaNova · 🇺🇸Austin, Texas, United States; San Jose, California, Unite…$131K - $197KJuniorPythonMachine LearningLLM
- Engineering Manager - ML/AI ProductsXero · 🇦🇺AU: Sydney (45 Clarence St)—HybridMachine LearningLLMAI
- Senior Software Engineer - ML & AI (AI Products)Xero · 🇦🇺AU: Sydney (45 Clarence St)—HybridSeniorGoPythonTypeScript
- Senior Engineer - ML & AI SystemsXero · 🇦🇺AU: Sydney (45 Clarence St)—HybridSeniorGoPythonTypeScript
- Staff Machine Learning Engineer, Sensors & IntelligenceTonal · 🇨🇦Toronto, ONCA$160K - CA$185KRemoteLead and aboveMachine LearningPyTorchTensorFlow
- Research Engineer, InfrastructureHarvey · 🇺🇸San Francisco · +3 more locations$230K - $450KHybridPythonPyTorchLLM evaluation
- Sr. Model Engineer, Post-Training & RLxAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Sr. Model Engineer, Mid-TrainingxAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Laboratory Technologist III, iPSC & Advanced Cellular ModelsThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut$61K - $88KJAX
- Senior Scientist, iPSC & Advanced Cellular Models R&DThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut$122K - $177KSeniorJAXPRDsProject management
- Research Assistant IIThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut$43K - $72KPythonRJAX
- Director Government and Community RelationsThe Jackson Laboratory's · 🇺🇸New York, New York—Lead and aboveJAXEmployee relationsRegulatory compliance
- Software Engineer Intern, Autonomy Software [Summer 2027]Zoox · 🇺🇸Foster City, CA—InternPythonC++Machine Learning
- Facilities Support and Grounds Technician II - Bar HarborThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine—Driver's licenseJAXPreventive maintenance
- Administrative Assistant - Research Support ServicesThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut—JAXSpreadsheets
- Bioinformatician I - Anczukow LabThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut$65K - $109KPythonMachine LearningBash
- Stationary Engineer II - Bar HarborThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine—JAXBlueprint readingPreventive maintenance
- Postdoctoral Fellow | Host-Microbiome Biology/ Organ-on-Chip Models | Jalili LabThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut—JAX
- Senior Robotics Engineer, Robot LearningSenra Systems · 🇺🇸Cypress, California, United States$200K - $260KSeniorPythonC++PyTorch
- Senior Software Engineer, Machine Learning PlatformAirwallex · 🇺🇸US - Seattle · +1 more location—HybridSeniorPythonJavaC++
- Senior Program ManagerThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine—SeniorJAX
- Scientist - Neurodegeneration ModelsThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine$86K - $144KJAXProject management
- Research Assistant II, Rare Disease Translational CenterThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine$43K - $72KJAX
- Study Director, Rare Disease Translational CenterThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine$90K - $131KLead and aboveJAXProject management
- Administrative Assistant III, JMCRSThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine—JAXPowerPointMicrosoft Word
- Software Engineer - ManipulationApptronik · 🇺🇸Sunnyvale, CA$180K - $240KPythonC++PyTorch
- AI Engineer - 11952Coupa · 🇨🇴Bogota, Colombia—PythonGCPTensorFlow
- Assistant, Associate and Full Professor | Farmington, CTThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut$130K - $159KJAX
- Assistant & Associate Professor | Bar Harbor, METhe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine—JAX
- Senior Scientist - Humanized Mouse and B Cell ImmunologyThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine$114K - $190KSeniorJAXProject management
- Machine Learning Scientist - Molecular Force FieldsSandboxaq · 🇺🇸United States$134K - $252KRemotePythonMachine LearningPyTorch
- Computational Chemist - Machine Learned Force Fields, Material DiscoverySandboxaq · 🇺🇸United States$134K - $252KRemotePythonMachine LearningPyTorch
- Machine Learning Scientist, Generative Molecular Design (Material Discovery)Sandboxaq · 🇺🇸United States$134K - $252KRemoteGCPMachine LearningLLM
- Machine Learning Scientist – Generative Molecular DesignSandboxaq · 🇺🇸United States$134K - $252KRemoteGCPMachine LearningLLM
- Autonomy Engineer Intern, Reinforcement Learning, 2027 (Zurich)Skydio · 🇨🇭Zurich, Switzerland—InternPythonC++Machine Learning
- Autonomy Engineer Intern, Reinforcement Learning, 2027 (HQ)Skydio · 🇺🇸San Mateo, California, United States—InternPythonC++Machine Learning
- Research Assistant II — Auditory NeuroscienceThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine$43K - $72KJAX
- 2027 Summer Intern, MS/PhD, Machine Learning, Driver Refinement FoundationsWaymo · 🇺🇸San Francisco, CA, USA · +1 more location—InternPythonC++SQL
- Photonic IC Design Engineer, Design Automation & ModelingK2 Space · 🇺🇸Seattle, WA$130K - $200KPythonMachine LearningPyTorch
- 2027 Summer Intern, MS/PhD, Software Engineer, ML Frameworks and EfficiencyWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningPyTorch
- Software Engineer - TritonGraphcore · 🇬🇧Bristol, UK · +1 more location—PythonC++Machine Learning
- Engineering Technician - Automation & Mechanical SystemsThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine · +1 more location—JAXMechatronicsCAD
- Mechanical Engineer - Research & DevelopmentThe Jackson Laboratory's · 🇺🇸Bar Harbor, Maine$80K - $116KPythonMATLABJAX
- Senior Machine Learning Engineer, Scenes, Traffic ControlsWaymo · 🇺🇸Mountain View, California, United States, San Francisco…$213K - $263KSeniorPythonC++SQL
- 2027 Summer Intern, PhD, Research, Multimodal Foundation ModelsWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningPyTorch
- 2027 Summer Intern, PhD, AI Foundations - RecipesWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningPyTorch
- Software EngineerWaymo · PERM - N/A$250K - $263KPythonC++Machine Learning
- Senior Scientist I, Computational Protein DesignGenerate Biomedicines · 🇺🇸Somerville, MA$172K - $241KSeniorPythonMachine LearningLLM
- Principal Data Scientist, User EcosystemReddit · 🇺🇸Remote - United States$313K - $365KRemoteLead and aboveMachine LearningPyTorchRecommender systems
- AI Performance Engineer - New GradSambaNova · 🇺🇸Austin, Texas, United States; San Jose, California, Unite…$131K - $197KJuniorPythonC++Machine Learning
- Graduate Software Engineer, AI/ ML Test Systems 2027Graphcore · 🇬🇧Bristol, UK—JuniorPythonC++Machine Learning
- 2027 Summer Intern, PhD, Research, VLM Harnesses for AV PlanningWaymo · 🇺🇸Mountain View, CA, USA—InternMachine LearningLLMPyTorch
- 2027 Summer Intern, PhD, Machine Learning - Simulation & Closed-Loop TrainingWaymo · 🇺🇸Mountain View, California, United States—InternPythonC++Machine Learning
- Staff Engineer, Hivemind Reasoning (R6292)Shield AI · 🇺🇸San Mateo, CA, US—Lead and abovePyTorchTensorFlowReinforcement learning
- Senior Manager, Engineering - Hivemind Reasoning (R6291)Shield AI · 🇺🇸San Mateo, CA, US—SeniorPyTorchTensorFlowReinforcement learning
- 2027 Summer Intern, PhD, Research Scientist, Visual ReasoningWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningLLM
- 2027 Summer Intern, PhD, Research, Behavior Post-trainingWaymo · 🇺🇸Mountain View, CA, USA; San Francisco, CA, USA—InternMachine LearningLLMPyTorch
- Software Engineer, ML Training OptimizationZoox · 🇺🇸Foster City, CA$177K - $270KPythonC++PyTorch
- Senior/Staff Software Engineer, ML Training OptimizationZoox · 🇺🇸Foster City, CA$254K - $350KLead and abovePythonC++PyTorch
- Senior Staff Software Engineer, StorageCrusoe · 🇺🇸Sunnyvale, CA - US$245K - $290KLead and aboveRustGoC++
- 2027 Summer Intern, MS/PhD, SemanticsWaymo · 🇺🇸Mountain View, California, USA—InternPythonMachine LearningPyTorch
- Senior Perception Learning EngineerApptronik · 🇺🇸Sunnyvale, CA$220K - $250KSeniorPythonC++PyTorch
- Postdoctoral Associate | Cardiac Physiology and Functional Genomics | Hinson LabThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut—JAX
- Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)Cohere · EuropeCA$295K - CA$535KRemoteLLMPyTorchReinforcement learning
- Clinical Research SpecialistThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut$70K - $101KGCPJAXClinical trials
- Senior Quality EngineerThe Jackson Laboratory's · 🇺🇸Farmington, Connecticut—SeniorJAXTechnical writingAudit
Showing the newest 100. See all 375 in search
Who is hiring for JAX roles
Where JAX roles are
- 🇺🇸United States271
- 🇺🇸California196
- 🇺🇸San Francisco Bay Area179
- 🇪🇺Europe58
- 🇬🇧United Kingdom39
- 🇬🇧London31
- 🌏Asia-Pacific25
- 🇺🇸New York, NY24
- 🇺🇸Seattle, WA19
- 🇨🇦Canada11
A role that lists several places is counted in each of them.
What JAX roles pay, by level
| Level in the title | Median advertised salary | Roles counted |
|---|---|---|
| No level in the title | $217K | 106 |
| Senior | $232K | 43 |
| Lead and above | $280K | 46 |
Medians of the midpoints of advertised yearly US-dollar ranges; a level needs at least 15 such roles. Advertised, not verified pay.
Jobs you can get with JAX
| Kind of work | Open jobs asking for JAX | Median advertised pay |
|---|---|---|
| Machine learning engineer | 160 | $242K |
| Research scientist | 86 | $284K |
| Software engineer | 68 | $275K |
| PhD and doctoral roles | 41 | – |
| Summer intern | 39 | – |
| ML infrastructure and MLOps | 34 | $250K |
| Reinforcement learning researcher | 27 | $275K |
| Computer vision engineer | 18 | $238K |
Skills asked together with JAX
PyTorch (81%), Python (70%), Machine Learning (62%), AI (39%), Deep learning (37%), TensorFlow (35%), C++ (33%), LLM (33%) of the 375 open postings that name JAX.
Companies hiring for JAX
JAX salariesQuestions about JAX jobs
- What is JAX?
- Google’s library for fast numerical computing and AI research. 375 open roles on Jobs Every Hour ask for it as of Nov 15, 2026.
- How many JAX jobs are open right now?
- There are 375 open JAX jobs at 111 companies on Jobs Every Hour as of Nov 15, 2026. 18 of them were posted in the last seven days. The list is rebuilt every hour from the employers' own careers pages.
- Which companies have the most JAX jobs?
- Right now the most openings are at Waymo (69), The Jackson Laboratory's (40), xAI (13), Anthropic (12) and Cohere (12). The ranking changes as companies open and close roles.
- Are there remote JAX jobs?
- 38 of the 375 open JAX jobs (10%) are listed as remote. Remote roles are often limited to certain countries or time zones, so check the location line of each job.
- What do JAX jobs pay?
- The median advertised salary is $238K, based on 198 open roles that publish a yearly US-dollar range. The middle half of those roles advertise between $193K and $288K. These are advertised ranges, not verified pay. Most employers here are technology companies, so the figures run above the wider market for the same work.
- Is it hard to get a JAX job?
- 57 of the 375 open JAX jobs (15%) are for beginners: junior roles and internships. Of the 144 postings that state experience, the median asks for 4 years. 111 companies are hiring and 18 roles were posted in the last seven days. Matching your resume to the skills each posting names helps most; the free resume builder shows the match.
- How is AI changing JAX jobs?
- 324 of the 375 open JAX jobs (86%) ask for AI or machine-learning skills, most often PyTorch, Machine Learning, AI. Employers still list 375 such roles today at 111 companies. Counted from the postings' own text on Nov 15, 2026.
- What skills do JAX jobs ask for?
- Most often JAX (100%), PyTorch (81%), Python (70%), Machine Learning (62%), AI (39%) of the 375 open postings, counted from their own text on Nov 15, 2026.
- Which job titles come up most in JAX jobs?
- 2027 Summer Intern (33), Member of Technical Staff (30), Research Engineer (25), Senior Machine Learning Engineer (17), Machine Learning Engineer (15): the most common titles among the 375 open roles, without the level and place notes in brackets.
- Are JAX jobs in demand in 2026?
- 375 JAX jobs are open on Jobs Every Hour today at 111 companies, 0.3% of all 148,162 open jobs here; 18 were posted in the last seven days. The median advertised salary is $238K. 38 (10%) can be done remotely.
- How do I apply for JAX jobs?
- Open a job and press “Apply on company site”. You go straight to the employer's own careers page and apply there. Jobs Every Hour needs no account and charges nothing for applying.
Narrow down JAX jobs
By place
- JAX jobs in the United States 271
- JAX jobs in California 196
- JAX jobs in San Francisco Bay Area 179
- JAX jobs in Europe 58
- JAX jobs in the United Kingdom 39
- JAX jobs in London 31
- 🌏JAX jobs in Asia-Pacific 25
- JAX jobs in New York, NY 24
- JAX jobs in Seattle, WA 19
- JAX jobs in Canada 11
- JAX jobs in Singapore 10
























