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

Lotus

lotus.ai
lotus.ai • HQ: Salt Lake City, United States • Medical AI • Telemedicine & Virtual Care • 201-500 employees • Startup • Trust B • Hiring 3
HQ: Salt Lake City, United States • Medical AI • Telemedicine & Virtual Care • 201-500 employees • Startup • Trust B • Hiring 3
Lotus AI gives you infinite personalized care powered by your health data, AI trained on leading medical evidence, and real doctors you can trust.
No named decision-maker yet.
Careers mailbox is known
Open 328 daysVerified live · 2 hours ago 11 months ago Long shot · 4%

Research Engineer

≈ $163k – $358k per year (Estimated) • In office (San Francisco, United States) • Full-Time
Python
SQL
AI/ML
LoRA
vLLM
Quantization
JAX
Function Calling
AI Agents
TensorRT
QLoRA
PyTorch
LLM
RAG
Hallucination
Synthetic Data
Red Teaming
LLM Evaluation
Agentic Workflows
Tool Use
RLHF
Multimodal AI
Knowledge Distillation
Langfuse
PEFT
Model Distillation
RLAIF
DevOps
Docker
Amazon ECS
QA
Sentry
+21
Apply

SambaNova Systems

sambanova.ai
sambanova.ai • HQ: Palo Alto, United States • LLM & Generative AI • Servers & Data Center Hardware • AI Compute & Inference • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2017 • Trust B • Hiring 55, 2 new
HQ: Palo Alto, United States • LLM & Generative AI • Servers & Data Center Hardware • AI Compute & Inference • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2017 • Trust B • Hiring 55, 2 new
SambaNova Systems is an American artificial intelligence company founded in 2017 by Stanford professors Kunle Olukotun and Christopher Re together with former Oracle executive Rodrigo Liang. It designs the Reconfigurable Dataflow Unit, a processor whose datapath is configured to match a model's computation graph rather than executing fixed instructions, and pairs it with a three-tier memory system that keeps very large models resident. Headquartered in Palo Alto, California, the company sells complete systems and a hosted cloud service to enterprises and governments that want to run frontier open models on their own terms rather than through a public model provider.
Kunle Olukotun · Co-Founder & Chief Technologist · decision-maker
Rodrigo Liang · Co-Founder & Chief Executive Officer · decision-maker
Christopher Re · Co-Founder · founder
Anthony Lin · Director
Careers mailbox is known
Open 331 daysTop 25% pay Visa sponsor 11 months ago Long shot · 4%

Senior AI Performance Engineer

$180k – $255k per year • Senior • 3+ years exp • In office (San Jose, United States) • Bachelor's Degree • Full-Time • Senior
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
Qwen
DeepSeek
Quantization
JAX
AI Agents
Llama
TensorFlow
PyTorch
Machine Learning
DeepSpeed
vLLM
CUDA Toolkit
Multimodal AI
TensorRT
OpenCL
LLM
CUDA
Triton
Megatron-LM
cuDNN
+16
Apply

InteractiveAI

interactive.ai
interactive.ai • HQ: Madrid, Spain • Custom Software Development • AI Agents • 11-50 employees • Est. 2025 • Likely startup • Hiring 3, 1 new
HQ: Madrid, Spain • Custom Software Development • AI Agents • 11-50 employees • Est. 2025 • Likely startup • Hiring 3, 1 new
The end-to-end AI environment where domain experts and developers build and operate Interactive Agents that run real work across your teams. Authored by your experts, governed every turn, and yours to own.
No named decision-maker yet.
Careers mailbox is known
Open 332 daysVerified live · 3 hours ago 11 months ago Long shot · 7%

Lead AI Engineer

$123k – $146k per year • Lead • 5+ years exp • Hybrid (Madrid, Spain) • Full-Time • Staff
Python
AI/ML
LangGraph
Spark
LlamaIndex
Dagster
Fine-tuning
Embeddings
JAX
AI Agents
TensorFlow
PyTorch
LLM
RAG
Agentic Workflows
LangChain
DevOps
GCP
Azure
CI/CD
AWS
Analytics
A/B Testing
+12
Apply

Synaptrix Labs

synaptrix-labs.com
synaptrix-labs.com • HQ: New York, United States • Custom Software Development • Machine Learning • Est. 2022 • Startup • Hiring 2
HQ: New York, United States • Custom Software Development • Machine Learning • Est. 2022 • Startup • Hiring 2
Introducing Atlas™ by Synaptrix, the first device that allows you to use your computer, wheelchair, prosthetics, and to communicate through brain activity alone.
Robert Crutchfield · Executive Officer, Director · decision-maker
Shyamy Sastry · Executive Officer, Director · decision-maker
Anita Watkins · Director
Kurt Jacobus · Director
Careers mailbox is known
Open 345 daysVerified live · 1 hour ago 11 months ago Long shot · 3%

Research Scientist, Machine Learning (PhD)

≈ $152k – $334k per year (Estimated) • Equity • In office (New York, United States) • PhD
Reinforcement Learning
JAX
PyTorch
Self-Supervised Learning
Time Series Forecasting
Machine Learning
Apply

Humble Robotics

humblerobotics.ai
humblerobotics.ai • HQ: San Francisco, United States • Road Transport • Autonomous Driving • 51-200 employees • Est. 2025 • Startup • Trust A • Hiring 21
HQ: San Francisco, United States • Road Transport • Autonomous Driving • 51-200 employees • Est. 2025 • Startup • Trust A • Hiring 21
Humble Robotics builds the Humble Hauler, a fully autonomous electric Class 8 truck for dock-to-dock freight transport. The vehicle has zero emissions and a 200-mile range, and open roles cover machine learning, embedded, electrical and commercial strategy positions. The company is headquartered in San Francisco, California, United States.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 11 min ago Visa sponsor 11 months ago Long shot · 16%

ML Software Engineer

$100k – $300k per year • Junior • 2+ years exp • San Francisco • Remote (United States) • Bachelor's Degree • Full-Time • Junior
Python
AI/ML
JAX
Multimodal AI
VLM
TensorFlow
PyTorch
LLM
Tokenization
Vision-Language-Action
Physical AI
DevOps
CI/CD
+3
Apply

jig.jp

jig.jp
jig.jp • HQ: Tokyo, Japan • Consumer & Utility Software • Likely startup • Trust B • Hiring 6
HQ: Tokyo, Japan • Consumer & Utility Software • Likely startup • Trust B • Hiring 6
jig.jp is a Japanese software company that develops consumer and peer-to-peer apps and services. Its main service is the live streaming platform Fuwacchi, alongside VTuber businesses, the topia service and the SABERA AR glasses, and it hires backend engineers, designers, product owners and new graduates. The company is based in Japan.
No named decision-maker yet.
No careers mailbox known
Open 347 daysVerified live · 4 hours ago 11 months ago Long shot · 3%

【福井勤務】エンジニア

In office (Fukui, Japan) • Full-Time
Java
JavaScript
TypeScript
Dart
Java
Spring Boot
AI/ML
MLFlow
JAX
PyTorch
Flax
Amazon SageMaker
Frontend
Next.js
Angular
Astro
React.js
Mobile
Flutter
DevOps
GitHub
GitLab
+9
Apply

Sunayu

sunayu.com
sunayu.com • HQ: Bethesda, United States • Cybersecurity • DevSecOps • DevOps & Platform Engineering • Hiring 25, 6 new
HQ: Bethesda, United States • Cybersecurity • DevSecOps • DevOps & Platform Engineering • Hiring 25, 6 new
Sunayu provides secure, agile, scalable, and dynamic DevSecOps and technical consulting solutions. Explore our services and industry expertise today!
No named decision-maker yet.
Careers mailbox is known
Open 350 daysVerified live · 1 hour ago 11 months ago Long shot · 3%

MD-Software Engineer(Java) - TS/SCI w/ Polygraph

≈ $127k – $269k per year (Estimated) • In office (Annapolis Junction, United States) • Bachelor's Degree • Full-Time • TS/SCI
Python
JavaScript
SQL
C++
Groovy
Java
Kotlin
TypeScript
C
Java
Spring Framework
Jackson
Maven
Apache Tomcat
Kotlin
Mockito
C
Embedded C
C++
CMake
Databases
PostgreSQL
AI/ML
JAX
Frontend
Angular
Bootstrap
JQuery
Mobile
JUnit
DevOps
Configuration Management
Windows
Rest API
OpenShift
Podman
CI/CD
Jenkins
Docker
Kubernetes
Linux
SOAP
Management
Agile
Scrum
QA
Selenium
+28
Apply

Substrate

substrate.com
substrate.com • HQ: San Francisco, United States • Semiconductor Manufacturing • Semiconductor Equipment • Est. 2022 • Startup • Trust B • Hiring 6
HQ: San Francisco, United States • Semiconductor Manufacturing • Semiconductor Equipment • Est. 2022 • Startup • Trust B • Hiring 6
Returning the United States to dominance in semiconductor production Supported by investors Extending Moore's Law into a new era of possibility.
No named decision-maker yet.
Careers mailbox is known
Open 355 daysTop 25% pay 12 months ago Long shot · 5%

Research Scientist, AI

$150k – $275k per year • Senior • 5+ years exp • In office (San Francisco, United States) • PhD • Full-Time • Senior
Python
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
Apply
co.bot • HQ: Santa Clara, United States • Collaborative Robots (Cobots) • Service & Hospitality Robotics • 51-200 employees • Startup • Trust A • Hiring 8
HQ: Santa Clara, United States • Collaborative Robots (Cobots) • Service & Hospitality Robotics • 51-200 employees • Startup • Trust A • Hiring 8
Collaborative Robotics is a robotics company that builds collaborative robots with predictable behaviours and real-world applicability. Its vacancies include deployment program managers, electrical, manufacturing and mechanical engineers, an embedded software engineer and an events and content marketing manager. The company is headquartered in Santa Clara, California, United States.
No named decision-maker yet.
Careers mailbox is known
Open 360 daysVerified live · 3 hours ago 12 months ago Long shot · 6%

AI Research Engineer

≈ $178k – $335k per year (Estimated) • Middle • 4+ years exp • In office (Seattle, United States) • Relocation • Master's Degree • Full-Time • Middle
Python
AI/ML
Reinforcement Learning
JAX
Diffusion Models
TensorFlow
PyTorch
Machine Learning
Multimodal AI
Robotics
Reinforcement Learning
Imitation Learning
+2
Apply

XPeng

xiaopeng.com
xiaopeng.com • HQ: Guangzhou, China • Autonomous Driving • Automotive Manufacturing • 1001-5000 employees • Est. 2014 • Trust B • Hiring 1003, 24 new
HQ: Guangzhou, China • Autonomous Driving • Automotive Manufacturing • 1001-5000 employees • Est. 2014 • Trust B • Hiring 1003, 24 new
XPeng is a Chinese smart electric vehicle maker headquartered in Guangzhou, Guangdong, developing its own driver-assistance software, AI chips, electric powertrains and in-car operating systems for models such as the P7, G6, G9, X9 and MONA M03. Founded in 2014 by He Xiaopeng and partners, it listed in New York in 2020 and Hong Kong in 2021, passed one million cumulative deliveries by the end of 2025, sells in about 60 countries, and is developing the IRON humanoid robot and flying cars through affiliate Aridge. It hires AI algorithm, autonomous driving, chip and vehicle engineers, product managers, and international sales and finance staff.
No named decision-maker yet.
Careers mailbox is known
Verified live · 14 min ago 12 months ago Long shot · 8%

Research Scientist(VLM 预训练 / 后训练)

In office (Shenzhen, China) • Full-Time • Chinese: C1
JAX
VLM
Transformers
PyTorch
LLM
FSDP
DevOps
SLI/SLO/SLA
Apply

Blue Rose Research

blueroseresearch.org
blueroseresearch.org • HQ: New York, United States • Think Tanks & Policy Research • Market Research • 11-50 employees • Trust B • Hiring 4
HQ: New York, United States • Think Tanks & Policy Research • Market Research • 11-50 employees • Trust B • Hiring 4
About Blue Rose Research helps campaigns make higher quality strategic decisionsby democratizing access to accurate measurement. What we do Solve fundamental problems plaguing the polling industry today byapplying state of the art tools from machine learning and Bayesianstatistics.
No named decision-maker yet.
Careers mailbox is known
Open 363 daysVerified live · 3 hours ago 12 months ago Long shot · 6%

Machine Learning Lead (LLM)

$165k – $210k per year • Lead • 6+ years exp • Remote (United States) • Staff
Python
SQL
AI/ML
Fine-tuning
Embeddings
JAX
TensorFlow
PyTorch
LLM
RAG
Agentic Workflows
Machine Learning
DevOps
Terraform
Docker
Kubernetes
+6
Apply

Simular

simular.ai
simular.ai • HQ: Palo Alto, United States • AI Research Labs • Multimodal AI • AI Agents • 51-200 employees • Est. 2023 • Trust C • Hiring 21
HQ: Palo Alto, United States • AI Research Labs • Multimodal AI • AI Agents • 51-200 employees • Est. 2023 • Trust C • Hiring 21
Simular is an artificial intelligence company based in Palo Alto, California, and was founded in 2023. The company develops autonomous AI agents, such as its Sai co-worker and S2 framework, which automate complex digital tasks by interacting directly with computer interfaces and web browsers. It operates as a venture-backed startup focusing on the research and deployment of computer-use models for knowledge workers and enterprise teams.
Li Ang · Executive Officer · decision-maker
No careers mailbox known
Likely inactiveVerified live · 2 hours ago Visa sponsor 1 year ago Long shot · 2%

PhD Research Intern

Intern • In office (Palo Alto, United States) • PhD • Internship • Intern
Python
AI/ML
Reinforcement Learning
JAX
Multimodal AI
Computer Vision
PyTorch
RAG
Computer Use
Machine Learning
+1
Apply
Likely inactiveVerified live · 2 hours ago Visa sponsor 1 year ago Long shot · 4%

Research Scientist

≈ $165k – $361k per year (Estimated) • In office (Palo Alto, United States) • PhD • Full-Time
Fine-tuning
Embeddings
Reinforcement Learning
JAX
Multimodal AI
Computer Vision
AI Agents
PyTorch
LLM
Post-training
Pre-training
Reward Modeling
Machine Learning
+5
Apply

Tzafon

tzafon.ai
tzafon.ai • San Francisco • Zurich • Tel Aviv • Machine Learning • Foundation Models • 1-10 employees • Startup • Trust B • Hiring 3
San Francisco • Zurich • Tel Aviv • Machine Learning • Foundation Models • 1-10 employees • Startup • Trust B • Hiring 3
Tzafon is advancing machine intelligence. Our team develops models and systems for autonomous agents interacting with the real world.
No named decision-maker yet.
Careers mailbox is known
Open 374 daysTop 25% pay 1 year ago Long shot · 9%

Member of Technical Staff - Foundations

$200k – $500k per year • Equity • Staff+ • Hybrid (San Francisco, United States) • Full-Time • Staff
Python
AI/ML
CUDA Toolkit
Fine-tuning
JAX
PyTorch
CUDA
Triton
Post-training
Pre-training
FSDP
TPU
vLLM
Synthetic Data
OpenAI
Anthropic
+7
Apply

Pear VC

pear.vc
pear.vc • HQ: San Francisco, United States • Venture Capital • 51-200 employees • Est. 2013 • Trust A • Hiring 94, 2 new
HQ: San Francisco, United States • Venture Capital • 51-200 employees • Est. 2013 • Trust A • Hiring 94, 2 new
Pear VC is a venture capital firm specializing in pre-seed and seed-stage investments in startups. Founded by Pejman Nozad and Mar Hershenson, the firm provides hands-on support in areas such as product development, recruiting, fundraising, and go-to-market strategy. Pear has backed companies including DoorDash, Gusto, Aurora Solar, Vanta, and Guardant Health.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 1 year ago Long shot · 15%

AI / ML Engineer - Known

≈ $174k – $328k per year (Estimated) • Middle • 3+ years exp • In office (San Francisco, United States) • Full-Time • Middle
Python
AI/ML
Embeddings
JAX
TensorFlow
PyTorch
LLM
Hugging Face
Human-in-the-Loop
LLM Guardrails
Agentic Workflows
Machine Learning
Reinforcement Learning
LLM Evaluation
Voice Agents
Analytics
A/B Testing
+7
Apply

Lovelace

lovelace.ai
lovelace.ai • HQ: Pittsburgh, United States • Machine Learning • Custom Software Development • 51-200 employees • Est. 2023 • Startup • Trust B • Hiring 2
HQ: Pittsburgh, United States • Machine Learning • Custom Software Development • 51-200 employees • Est. 2023 • Startup • Trust B • Hiring 2
Lovelace is the only provider of enterprise-scale context engines that enable autonomous AI agents at the speed, scale, and accuracy required for mission-critical analysis.
Andrew Moore · Executive Officer, Director · decision-maker
William Porteous · Director
Elizabeth Stein · Director
No careers mailbox known
Open 375 daysVerified live · 22 min ago 1 year ago Long shot · 5%

Machine Learning Engineer

≈ $126k – $275k per year (Estimated) • In office (Pittsburgh, United States) • Bachelor's Degree • Full-Time
JAX
PyTorch
Knowledge Graph
XLA
Machine Learning
AI Agents
DevOps
GCP
Azure
AWS
+1
Apply

METISINC

metisinc.io
metisinc.io • HQ: San Francisco, United States • IT Consulting & Digital Transformation • Custom Software Development • Est. 2025 • Trust B • Hiring 4
HQ: San Francisco, United States • IT Consulting & Digital Transformation • Custom Software Development • Est. 2025 • Trust B • Hiring 4
Asymmetrical Solutions Realisation & Asynchronous Delivery. Bespoke business design andengineering. Advisory, consulting, training, & recruitment.
No named decision-maker yet.
No careers mailbox known
Open 377 daysVerified live · 1 hour ago 1 year ago Long shot · 3%

Research Engineer

$200k per year • In office (San Francisco, United States)
Python
AI/ML
Reinforcement Learning
JAX
AI Agents
TensorFlow
PyTorch
Synthetic Data
OpenAI
Post-training
Reward Modeling
Machine Learning
DevOps
Platform Engineering
+4
Apply

Pinely

pinely.com
pinely.com • HQ: Amsterdam, Netherlands • Online Brokerage & Trading Platforms • Trust A • Hiring 17, 2 new
HQ: Amsterdam, Netherlands • Online Brokerage & Trading Platforms • Trust A • Hiring 17, 2 new
Pinely is a privately owned & funded algorithmic trading firm, based in Singapore, The Netherlands and Cyprus, and growing by leaps and bounds. We focus on high frequency and ultra low latency trading.
No named decision-maker yet.
Careers mailbox is known
Open 388 daysVerified live · 4 hours ago 1 year ago Long shot · 3%

ML Performance Engineer

≈ $61k – $129k per year (Estimated) • In office (Amsterdam, Netherlands) • Master's Degree • Full-Time
Python
AI/ML
CUDA Toolkit
JAX
Transformers
PyTorch
CUDA
Triton
Apply

Collate

collate.com
collate.com • HQ: San Francisco, United States • Clinical Trial Software • Document AI • 51-200 employees • Startup • Trust D • Hiring 11
HQ: San Francisco, United States • Clinical Trial Software • Document AI • 51-200 employees • Startup • Trust D • Hiring 11
Our mission Accelerate life-saving innovations by creating accurate documentation for the life sciences industry. Catalyze progress Automate paperwork at every stage of development and commercialization.
No named decision-maker yet.
No careers mailbox known
Likely inactiveVerified live · 2 hours ago 1 year ago Long shot · 4%

AI Research Scientist

≈ $163k – $358k per year (Estimated) • In office (San Francisco, United States)
JAX
NLP
TensorFlow
PyTorch
LLM
Machine Learning
Apply

Mantari

mantari.com
mantari.com • HQ: Torrance, United States • Geospatial AI • Energy AI • Startup • Trust B • Hiring 6
HQ: Torrance, United States • Geospatial AI • Energy AI • Startup • Trust B • Hiring 6
Earth, Quantified.. Weather drives advantage, delivering clarity for commerce, government, national security, and civilization itself.
No named decision-maker yet.
Careers mailbox is known
Open 402 daysVerified live · 17 min ago 1 year ago Long shot · 3%

Software Engineer

≈ $120k – $253k per year (Estimated) • In office (Torrance, United States)
Python
Rust
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
CUDA Toolkit
JAX
TensorFlow
PyTorch
Ray
CUDA
DevOps
GCP
SLURM
Azure
CI/CD
AWS
Kubernetes
+9
Apply

Alinia

alinia.ai
alinia.ai • HQ: Dover, United States • AI Safety & Alignment • AI Agents • 11-50 employees • Est. 2023 • Startup • Hiring 2
HQ: Dover, United States • AI Safety & Alignment • AI Agents • 11-50 employees • Est. 2023 • Startup • Hiring 2
We are on a mission to make human-machine interactions safer in the most critical scenarios and at scale. Our core values At Alinia, we are all driven by a passion to make people’s lives better. We approach challenges with a growth mindset and strive to amplify others.
Ariadna Font Llitjos · Executive Officer, Director · decision-maker
Carlos Munoz Ferrandis · Executive Officer, Director · decision-maker
Manuel Silva Martinez · Director
Careers mailbox is known
Open 403 daysVerified live · 1 hour ago 1 year ago Long shot · 7%

Applied Research Scientist

≈ $48k – $112k per year (Estimated) • Hybrid (Barcelona, Spain) • Master's Degree • Full-Time
Python
AI/ML
JAX
NLP
TensorFlow
PyTorch
LLM
Data Augmentation
Synthetic Data
SFT
LLM Guardrails
Machine Learning
Post-training
+4
Apply
Machines can read this selection · checked 23:45 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.