JAX Jobs - Remote & On-site

JAX jobs hiring now at vetted startups and product companies on Alion, updated daily. Average salary around $276,623/year. Compare remote, hybrid and on-site positions with relocation and visa sponsorship. Browse and apply today.

AI/ML
Software
Data
Security
DevOps & Cloud
Leadership
Product
Design
Fintech
Hardware & Chips
Defense
Aerospace
Biotech
Energy
Robotics
Automotive
MedTech
Science

Applied Intuition

appliedintuition.com
appliedintuition.com • HQ: Sunnyvale, United States • Defense AI • Automotive Software • Simulation & Digital Twin Software • Autonomous Driving • 1001-5000 employees • Est. 2017 • Trust B • Hiring 290, 5 new
HQ: Sunnyvale, United States • Defense AI • Automotive Software • Simulation & Digital Twin Software • Autonomous Driving • 1001-5000 employees • Est. 2017 • Trust B • Hiring 290, 5 new
Applied Intuition is an American vehicle software company founded in 2017 that sells the toolchain automakers use to develop, simulate and validate autonomous and assisted driving systems. Its products cover large-scale simulation, sensor modelling, drive-data management, scenario generation and a vehicle operating system, sold to carmakers, truck, agricultural and construction equipment manufacturers and defence customers, and it has expanded into defence autonomy through acquisitions. The company is headquartered in Sunnyvale, California, United States.
Peter Ludwig · Co-founder and Chief Technology Officer · decision-maker
Qasar Younis · Co-founder and Chief Executive Officer · decision-maker
Careers mailbox is known
Likely inactiveTop 25% pay Visa sponsor 1 year ago Long shot · 6%

ML Runtime Optimization Engineer

$217k – $318k per year • Equity • Middle • 3+ years exp • In office (Sunnyvale, United States) • Bachelor's Degree • Full-Time • Middle
Quantization
JAX
ONNX
TensorRT
CUDA
Triton
XLA
Physical AI
PyTorch
+1
Apply

Maven Robotics

mavenrobotics.ai
mavenrobotics.ai • HQ: San Francisco, United States • Robotics AI • Humanoid Robots • 51-200 employees • Startup • Trust A • Hiring 16, 5 new
HQ: San Francisco, United States • Robotics AI • Humanoid Robots • 51-200 employees • Startup • Trust A • Hiring 16, 5 new
Maven Robotics is a robotics company building general-purpose AI robots and currently operating in stealth. It hires engineers in motion planning and controls, robot perception and manipulation, embedded real-time software and MLOps. The company is headquartered in San Francisco, California, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago Visa sponsor 1 year ago Long shot · 23%

Robotics AI Research Engineer

≈ $124k – $298k per year (Estimated) • In office (Santa Clara, United States) • Master's Degree
Python
AI/ML
Reinforcement Learning
JAX
Multimodal AI
Vision-Language-Action
World Models
PyTorch
Machine Learning
Robotics
Reinforcement Learning
+1
Apply

Xiaomi

mi.com
mi.com • HQ: Beijing, China • Televisions • Automotive Manufacturing • Smart Home Devices • Smartphones • 1001-5000 employees • Est. 2010 • Trust C • Hiring 1084, 42 new
HQ: Beijing, China • Televisions • Automotive Manufacturing • Smart Home Devices • Smartphones • 1001-5000 employees • Est. 2010 • Trust C • Hiring 1084, 42 new
Xiaomi is a Chinese consumer electronics company founded in Beijing in 2010 by Lei Jun and seven co-founders, and one of the three largest smartphone makers in the world. It sells Xiaomi and Redmi handsets alongside a very broad connected-device ecosystem covering televisions, wearables, scooters, appliances and home sensors, built partly through investments in dozens of partner manufacturers. Since 2021 the company has added electric vehicles, launching the SU7 saloon and YU7 sport utility to unexpectedly strong demand, and it ties the whole portfolio together with its HyperOS operating system.
Lin Bin · Co-Founder & Vice Chairman · decision-maker
Lei Jun · Co-Founder, Chairman & Chief Executive Officer · decision-maker
Alain Lam · Chief Financial Officer · decision-maker
Lu Weibing · President · decision-maker
No careers mailbox known
Likely inactiveVerified live · 3 hours ago 2 years ago Long shot · 5%

大模型训练与推理Infra工程师-MiMo

In office (Beijing, China) • Full-Time • Chinese: C1
C++
Python
C++
TensorFlow C++
AI/ML
DeepSpeed
JAX
Transformers
Ray
CUDA
NCCL
cuDNN
TensorFlow
PyTorch
DevOps
Terraform
Docker
Kubernetes
+7
Apply

Droyd Robotics

droydrobotics.com
droydrobotics.com • HQ: San Francisco, United States • Construction Robotics • Startup • Trust D • Hiring 23
HQ: San Francisco, United States • Construction Robotics • Startup • Trust D • Hiring 23
Tell Us About Your Environment. Drop us a note about your construction environment or your operations and maintenance pain points.
No named decision-maker yet.
No careers mailbox known
Likely inactiveTop 25% pay 2 years ago Long shot · 2%

Machine Learning Researcher

$150k – $400k per year • In office (San Francisco, United States) • Master's Degree • Full-Time
Fine-tuning
JAX
Embodied AI
PyTorch
Apply
Likely inactiveVerified live · 1 hour ago 2 years ago Long shot · 2%

Machine Learning Intern

≈ $53k – $87k per year (Estimated) • Intern • In office (San Francisco, United States) • Internship • Intern
JAX
PyTorch
Machine Learning
Apply

Terra Quantum

terraquantum.swiss
terraquantum.swiss • HQ: San Francisco, United States • Quantum Science & Computing • 51-200 employees • Trust B • Hiring 7
HQ: San Francisco, United States • Quantum Science & Computing • 51-200 employees • Trust B • Hiring 7
Terra Quantum offers Quantum as a Service in the areas of quantum algorithms, quantum computing and quantum security.
No named decision-maker yet.
No careers mailbox known
Open 894 daysVerified live · 1 hour ago 2 years ago Long shot · 8%

Senior Tensor AI Scientist

≈ $84k – $184k per year (Estimated) • Senior • 5+ years exp • Hybrid (Munich, Germany) • PhD • Senior
C++
Python
AI/ML
Fine-tuning
JAX
Knowledge Distillation
Model Distillation
PyTorch
Machine Learning
DevOps
HPC
+1
Apply

Spellbrush

spellbrush.com
spellbrush.com • HQ: San Francisco, United States • Game Development • Generative Image • Est. 2018 • Startup • Trust D • Hiring 10
HQ: San Francisco, United States • Game Development • Generative Image • Est. 2018 • Startup • Trust D • Hiring 10
Spellbrush builds generative models for anime art and the games made with them, including the illustration tool used by its community. Its research focuses on stylised character generation rather than photorealism. The studio ships its own titles alongside the underlying model work.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveVerified live · 3 hours ago Visa sponsorship 2 years ago Long shot · 2%

AI Anime Researcher

In office (Tokyo, Japan, San Francisco, United States) • Full-Time • Japanese: C1
JAX
Diffusion Models
TPU
PyTorch
Apply

Anlatan

anlatan.ai
anlatan.ai • LLM & Generative AI • Trust B • Hiring 3
LLM & Generative AI • Trust B • Hiring 3
Anlatan is an AI company that makes the NovelAI storytelling product, working on cutting-edge AI with a focus on user-friendly, privacy-safe and censorship-free products. It describes itself as a small team of personalities and hires frontend and backend engineers, machine learning engineers and UI designers. The company is based in the United States.
No named decision-maker yet.
Careers mailbox is known
Open 1104 daysVerified live · 1 hour ago 3 years ago Long shot · 5%

Machine Learning Engineer

≈ $135k – $299k per year (Estimated) • Remote (United States) • Full-Time
Python
AI/ML
Fine-tuning
NLP
PyTorch
Machine Learning
JAX
Triton
DevOps
SLURM
Kubernetes
+1
Apply

Reka AI

reka.ai
reka.ai • HQ: Sunnyvale, United States • LLM & Generative AI • Multimodal AI • 201-500 employees • Est. 2022 • Startup • Trust B • Hiring 6
HQ: Sunnyvale, United States • LLM & Generative AI • Multimodal AI • 201-500 employees • Est. 2022 • Startup • Trust B • Hiring 6
Reka AI is a research laboratory building multimodal foundation models that process text, images, audio and video in one system. Its Core and Flash models were trained from scratch by a small team of researchers previously at DeepMind, Google Brain, Baidu and Meta, and they compete with far larger laboratories on multimodal benchmarks. The company sells its models to enterprises and governments that want capable multimodal systems without depending on the largest providers; its models are available through an API and through cloud marketplaces for enterprise deployment.
Dani Yogatama · Executive Officer, Director · decision-maker
Cyprien d'Autume · Director
Robert Toews · Director
Careers mailbox is known
Open 1180 daysVerified live · 3 hours ago Visa sponsorship 3 years ago Long shot · 7%

Member of Technical Staff (ML)

≈ $140k – $269k per year (Estimated) • Staff+ • Remote (United States, United Kingdom) • Full-Time • Staff
JAX
Multimodal AI
PyTorch
Machine Learning
Apply

Neuralink

neuralink.com
neuralink.com • HQ: Fremont, United States • Medical & Healthcare Robotics • Neurotech & Neuromodulation • 1001-5000 employees • Est. 2016 • Trust B • Hiring 77, 1 new
HQ: Fremont, United States • Medical & Healthcare Robotics • Neurotech & Neuromodulation • 1001-5000 employees • Est. 2016 • Trust B • Hiring 77, 1 new
Neuralink is an American neurotechnology company founded in 2016 by Elon Musk and a team of scientists and engineers to build implantable brain-computer interfaces. Its N1 implant records neural activity through flexible electrode threads placed by a purpose-built surgical robot, allowing trial participants to control computers and devices directly with their thoughts. Headquartered in Fremont, California with a second campus in Austin, Texas, the company runs clinical studies aimed at restoring autonomy to people living with paralysis and is developing a vision prosthesis called Blindsight.
Max Hodak · Co-Founder · decision-maker
DJ Seo · Co-Founder & President · decision-maker
Elon Musk · Co-Founder & Chief Executive Officer · decision-maker
Jared Birchall · Executive Officer, Director · decision-maker
Careers mailbox is known
Likely inactiveVerified live · 2 hours ago Visa sponsor 3 years ago Long shot · 3%

Machine Learning Engineer

≈ $175k – $386k per year (Estimated) • Equity • In office (Austin, South San Francisco, United States) • Bachelor's Degree
Rust
Swift
TypeScript
Python
AI/ML
Reinforcement Learning
Quantization
Self-Supervised Learning
Time Series
SFT
PPO
Post-training
Pre-training
Machine Learning
JAX
PyTorch
Mobile
SwiftUI
Robotics
Sensor Fusion
Reinforcement Learning
+10
Apply

AE Studio

ae.studio
ae.studio • HQ: Los Angeles, United States • AI Consulting & Integration • Custom Software Development • 11-50 employees • Est. 2015 • Trust B • Hiring 6
HQ: Los Angeles, United States • AI Consulting & Integration • Custom Software Development • 11-50 employees • Est. 2015 • Trust B • Hiring 6
AE Studio is a software development and data science consultancy that reinvests a share of its revenue into artificial intelligence alignment research rather than distributing it to outside shareholders. Its alignment team works on neglected approaches to making advanced models safe, including interpretability, attention schema theory and brain computer interface research, and publishes the results openly. The company funds that work through client engineering projects and an internal incubator that has produced several independent product ventures.
No named decision-maker yet.
Careers mailbox is known
Open 2021 daysVerified live · 4 hours ago 5 years ago Long shot · 5%

Applied AI Data Scientist

≈ $135k – $298k per year (Estimated) • Remote (United States, Brazil) • English: B2
Python
AI/ML
Fine-tuning
JAX
NLP
Time Series
TensorFlow
Computer Vision
Cursor
PyTorch
RAG
LLM
Machine Learning
Management
Kanban
Scrum
Agile
+7
Apply
Open 2021 daysVerified live · 4 hours ago 5 years ago Long shot · 5%

Applied AI Data Scientist (Brazil)

Florianópolis • Remote (Brazil) • English: B2
Python
AI/ML
Fine-tuning
JAX
NLP
Time Series
TensorFlow
Computer Vision
Cursor
PyTorch
RAG
LLM
Machine Learning
Management
Kanban
Scrum
Agile
+7
Apply
Machines can read this selection · checked 05:19 UTC
JAX Jobs - Remote & On-site
Frequently asked questions
How many JAX jobs are available on Alion?
There are currently 525 active listings from 241 hiring companies on Alion, refreshed daily.
What salary do JAX jobs pay?
The average salary is around $276,623 per year based on the current listings.
Are remote options available for JAX jobs?
Yes, many of these roles offer remote or hybrid work, and you can filter listings by remote, relocation and visa sponsorship.
How do I apply for JAX jobs?
Open any listing on Alion and apply directly with the employer through the provided link - no lengthy sign-up required.