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
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Leadership
Networking
Product
Design
Analytics
Support
Solutions
Enterprise Apps
Quantum

Netic

netic.ai
netic.ai • HQ: San Francisco, United States • Conversational AI • Field Service Management Software • Marketing & Sales AI • 51-200 employees • Startup • Trust C • Hiring 27, 2 new
HQ: San Francisco, United States • Conversational AI • Field Service Management Software • Marketing & Sales AI • 51-200 employees • Startup • Trust C • Hiring 27, 2 new
Netic’s AI-driven platform helps service enterprises boost revenue, engage customers, and scale growth with intelligent automation. See how our multi-channel engine empowers your business.
No named decision-maker yet.
No careers mailbox known
Likely inactiveVerified live · 1 hour ago 1 year ago Long shot · 4%

Applied AI Research Engineer

≈ $163k – $357k per year (Estimated) • In office (San Francisco, United States) • Full-Time
Databricks
AI/ML
JAX
Multimodal AI
AI Agents
PyTorch
Feature Store
LLM Guardrails
Analytics
ETL/ELT
Apply

Character.AI

character.ai
character.ai • HQ: Redwood City, United States • Foundation Models • Conversational AI • LLM & Generative AI • 51-200 employees • Est. 2021 • Startup • Trust D • Hiring 13
HQ: Redwood City, United States • Foundation Models • Conversational AI • LLM & Generative AI • 51-200 employees • Est. 2021 • Startup • Trust D • Hiring 13
Character.AI is an American company founded in 2021 by former Google researchers that operates a platform for creating and talking to personality-driven AI characters. Users build characters with a description, greeting and example dialogue, then share them publicly, and the service has grown into one of the most used consumer AI products by session time. Headquartered in Redwood City, California, the company runs its own dialogue-tuned models and has added voice calls, animated avatars and multi-character scenes, alongside age assurance and safety controls for younger users.
Karandeep Anand · Chief Executive Officer · decision-maker
Daniel De Freitas · Co-Founder · founder
Noam Shazeer · Co-Founder · founder
Dominic Perella · former Interim Chief Executive Officer
Careers mailbox is known
Likely inactiveTop 25% pay 1 year ago Long shot · 4%

Machine Learning Infrastructure Engineer

$150k – $350k per year • Middle • 4+ years exp • Hybrid (Redwood City, United States) • Full-Time • Middle
JAX
TensorFlow
PyTorch
DevOps
Kubernetes
Apply

Gauss Labs

gausslabs.ai
gausslabs.ai • HQ: Seoul, South Korea • Semiconductor Manufacturing • Industrial AI • Machine Learning • 51-200 employees • Est. 2020 • Trust B • Hiring 7, 1 new
HQ: Seoul, South Korea • Semiconductor Manufacturing • Industrial AI • Machine Learning • 51-200 employees • Est. 2020 • Trust B • Hiring 7, 1 new
Gauss Labs is an industrial artificial intelligence company founded in 2020 by SK hynix to apply machine learning to semiconductor manufacturing. Its products improve metrology, defect detection and yield analysis inside fabrication plants. The company operates research teams in Korea and the United States.
No named decision-maker yet.
Careers mailbox is known
Open 542 daysVerified live · 1 hour ago Visa sponsor 1 year ago Long shot · 6%

AI Scientist - Machine Learning (KR/US)

≈ $159k – $300k per year (Estimated) • Middle • 3+ years exp • Hybrid (Palo Alto, United States) • PhD • Full-Time • Middle
Python
AI/ML
JAX
Transformers
TensorFlow
PyTorch
Time Series Forecasting
Machine Learning
Apply

MiniMax

minimax.io
minimax.io • HQ: Shanghai, China • Generative Audio & Music • Conversational AI • LLM & Generative AI • Foundation Models • Generative Video • 1001-5000 employees • Est. 2021 • Trust D • Hiring 145, 3 new
HQ: Shanghai, China • Generative Audio & Music • Conversational AI • LLM & Generative AI • Foundation Models • Generative Video • 1001-5000 employees • Est. 2021 • Trust D • Hiring 145, 3 new
MiniMax is a Chinese artificial intelligence company founded in Shanghai in 2021 by former SenseTime researchers, and one of the country's most commercially aggressive model developers. It builds text, speech, image and video foundation models in house and pushes them straight into consumer products, including the Hailuo video generator and the Talkie and Xingye companion applications that have found large audiences inside and outside China. The company also sells API access to enterprises, released the open-weight M-series reasoning models, and listed in Hong Kong in 2025.
Yan Junjie · Founder & Chief Executive Officer · decision-maker
Zhou Yucong · Co-Founder · founder
Yang Bin · Co-Founder · founder
No careers mailbox known
Likely inactiveVerified live · 1 hour ago 1 year ago Long shot · 2%

AI Infra平台研发工程师

In office (Shanghai, China) • Full-Time • Chinese: C1
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
JAX
TensorFlow
PaddlePaddle
PyTorch
Apply
Likely inactiveVerified live · 1 hour ago 3 years ago Long shot · 2%

大模型训练框架研发工程师-生成模型方向

In office (Beijing, China) • Full-Time • Chinese: C1
DeepSpeed
JAX
AI Agents
PyTorch
Mixture of Experts
Megatron-LM
NCCL
XLA
KV Cache
+1
Apply

Maple

maple.inc
maple.inc • HQ: New York, United States • Artificial Intelligence • Branding • Speech & Audio AI • Trust D • Hiring 14
HQ: New York, United States • Artificial Intelligence • Branding • Speech & Audio AI • Trust D • Hiring 14
Maple provides AI phone answering for restaurants. Voice covers answering, FAQs and call transfer; Pro adds ordering into supported POS systems and reservations. Drive-thru and kiosk voice ordering are in early access.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveTop 25% pay 1 year ago Long shot · 1%

ML Research Engineer

$120k – $250k per year • In office (New York, United States) • Bachelor's Degree • Full-Time
CUDA Toolkit
Reinforcement Learning
JAX
Prompt Engineering
AI Agents
NLP
Speech Recognition
PyTorch
RAG
CUDA
Triton
Text-to-Speech
TPU
Human-in-the-Loop
Knowledge Graph
Voice Agents
DevOps
CI/CD
+9
Apply

Oddin

oddin.gg
oddin.gg • Lima • Prague • Aarhus • São Paulo • iGaming & Gambling • Esports • Est. 2018 • Likely startup • Trust A • Hiring 23
Lima • Prague • Aarhus • São Paulo • iGaming & Gambling • Esports • Est. 2018 • Likely startup • Trust A • Hiring 23
Complete esports betting infrastructure for operators. Boost engagement and revenue via low latency odds feed, risk management, and unique solutions.
No named decision-maker yet.
Careers mailbox is known
Open 557 daysVerified live · 9 min ago 1 year ago Long shot · 2%

Research Scientist for Speech Synthesis

≈ $58k – $124k per year (Estimated) • Hybrid • Freelance
Python
AI/ML
NLP
Text-to-Speech
Edge AI
JAX
Diffusion Models
TensorFlow
PyTorch
Machine Learning
DevOps
CI/CD
+2
Apply

Anchr software

anchrsoftware.com
anchrsoftware.com • Breda • Mechelen • Evergem • Investment Holding Companies • Trust B • Hiring 7
Breda • Mechelen • Evergem • Investment Holding Companies • Trust B • Hiring 7
Anchr biedt softwarebedrijven een stevige basis om te groeien. Als house of companies verbinden we ondernemers die bouwen aan innovatieve B2B-software.
No named decision-maker yet.
Careers mailbox is known
Open 585 daysVerified live · 16 min ago 1 year ago Long shot · 9%

Technical Solution Architect

≈ $70k – $126k per year (Estimated) • Staff+ • In office (Breda, Netherlands) • Architect
JavaScript
Java
TypeScript
Java
Hibernate
Databases
PostgreSQL
RabbitMQ
ElasticSearch
AI/ML
JAX
Frontend
Vue.js
Angular
JQuery
DevOps
Rest API
Azure DevOps
Git
Docker
SOAP
Azure
Analytics
Informatica
+10
Apply

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 292, 18 new
HQ: Sunnyvale, United States • Defense AI • Automotive Software • Simulation & Digital Twin Software • Autonomous Driving • 1001-5000 employees • Est. 2017 • Trust B • Hiring 292, 18 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
Open 591 daysTop 25% pay Visa sponsor 1 year ago Long shot · 4%

ML Runtime Optimization Engineer

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

Maven Robotics

mavenrobotics.ai
mavenrobotics.ai • HQ: San Francisco, United States • Robotics AI • Humanoid Robots • 51-200 employees • Startup • Trust A • Hiring 13
HQ: San Francisco, United States • Robotics AI • Humanoid Robots • 51-200 employees • Startup • Trust A • Hiring 13
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 · 2 hours ago Visa sponsor 1 year ago Fading · 28%

Robotic AI Engineer/Applied Scientist - Foundation Models

≈ $133k – $319k per year (Estimated) • In office (San Francisco, United States) • Master's Degree
Python
AI/ML
Fine-tuning
Reinforcement Learning
JAX
Multimodal AI
PyTorch
Vision-Language-Action
World Models
Physical AI
Machine Learning
Robotics
Reinforcement Learning
+3
Apply

Xiaomi

mi.com
mi.com • HQ: Beijing, China • Televisions • Automotive Manufacturing • Smart Home Devices • Smartphones • 1001-5000 employees • Est. 2010 • Trust C • Hiring 966, 26 new
HQ: Beijing, China • Televisions • Automotive Manufacturing • Smart Home Devices • Smartphones • 1001-5000 employees • Est. 2010 • Trust C • Hiring 966, 26 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 · 1 hour ago 2 years ago Long shot · 3%

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

In office (Beijing, China) • Full-Time • Chinese: C1
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
DeepSpeed
CUDA Toolkit
JAX
Transformers
TensorFlow
PyTorch
Ray
CUDA
NCCL
cuDNN
DevOps
Terraform
Docker
Kubernetes
+9
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
PyTorch
Embodied AI
Apply

Terra Quantum

terraquantum.swiss
terraquantum.swiss • HQ: San Francisco, United States • Quantum Science & Computing • 51-200 employees • Trust B • Hiring 8
HQ: San Francisco, United States • Quantum Science & Computing • 51-200 employees • Trust B • Hiring 8
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 886 daysVerified live · 1 hour ago 2 years ago Long shot · 8%

Senior Tensor AI Scientist

≈ $84k – $171k per year (Estimated) • Senior • 5+ years exp • Hybrid (Munich, Germany) • PhD • Senior
Python
C++
C++
PyTorch C++
AI/ML
Fine-tuning
JAX
Knowledge Distillation
PyTorch
Model Distillation
Machine Learning
DevOps
HPC
+2
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 · 10 min 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
PyTorch
TPU
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 1096 daysVerified live · 1 hour ago 3 years ago Long shot · 5%

Machine Learning Engineer

≈ $137k – $300k per year (Estimated) • Remote (United States) • Full-Time
Python
AI/ML
Fine-tuning
NLP
PyTorch
Edge AI
Machine Learning
JAX
Triton
DevOps
SLURM
Kubernetes
+2
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 1172 daysVerified live · 2 hours ago Visa sponsorship 3 years ago Long shot · 7%

Member of Technical Staff (ML)

≈ $138k – $265k per year (Estimated) • Staff+ • Remote (United States, United Kingdom) • Full-Time • Staff
JAX
Multimodal AI
PyTorch
Edge AI
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 73, 2 new
HQ: Fremont, United States • Medical & Healthcare Robotics • Neurotech & Neuromodulation • 1001-5000 employees • Est. 2016 • Trust B • Hiring 73, 2 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

≈ $176k – $385k per year (Estimated) • Equity • In office (Austin, South San Francisco, United States) • Bachelor's Degree
Python
Rust
TypeScript
Swift
AI/ML
Reinforcement Learning
Quantization
Self-Supervised Learning
Time Series Forecasting
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 2013 daysVerified live · 2 hours ago 5 years ago Long shot · 5%

Applied AI Data Scientist

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

Applied AI Data Scientist (Brazil)

Florianópolis • Remote (Brazil) • English: B2
Python
AI/ML
Cursor
Fine-tuning
JAX
Computer Vision
NLP
TensorFlow
PyTorch
LLM
RAG
Time Series Forecasting
Machine Learning
Management
Agile
Scrum
Kanban
+7
Apply
Machines can read this selection · checked 04:17 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.