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

Wells Fargo

wellsfargo.com
wellsfargo.com · HQ: San Francisco, United States · Investment Banking & M&A Advisory · Wealth Management & Financial Advisors · Mortgage Lending & Brokers · Commercial & Retail Banks · 1001-5000 employees · Est. 1852 · Trust A · Hiring 2838, 387 new
HQ: San Francisco, United States · Investment Banking & M&A Advisory · Wealth Management & Financial Advisors · Mortgage Lending & Brokers · Commercial & Retail Banks · 1001-5000 employees · Est. 1852 · Trust A · Hiring 2838, 387 new
Headquartered in San Francisco, California, Wells Fargo & Company is one of the largest diversified financial services and banking institutions in the United States. The enterprise offers a broad suite of consumer banking, commercial lending, wealth management, and mortgage products to millions of individual and corporate clients. Operating an extensive network of physical branches alongside robust digital channels, it delivers critical financial infrastructure across domestic and international markets.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago Visa sponsorship 1 month ago Long shot · 9%

Executive Director (Senior Lead Securities Python Quantitative Developer )

≈ $150k – $296k per year (Estimated) · Head+ · 7+ years exp · In office (Charlotte, United States) · Master's Degree · Full-Time · Staff
C++
Python
Cython
Java
Cython
PyBind11
AI/ML
Numba
JAX
CUDA
Triton
XLA
AI Agents
DevOps
Git
Management
Agile
Confluence
Jira
+7
Apply

InMobi

inmobi.com
inmobi.com · HQ: Bengaluru, India · Advertising · 501-1000 employees · Est. 2007 · Trust A · Hiring 60, 7 new
HQ: Bengaluru, India · Advertising · 501-1000 employees · Est. 2007 · Trust A · Hiring 60, 7 new
InMobi is a global adtech and consumer technology multinational headquartered in Bengaluru, India, with a major corporate presence in Singapore, San Francisco, and London. Founded in 2007 (originally as mKhoj), it made history in 2011 by becoming India's first unicorn startup.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago Visa sponsor 1 month ago Likely open · 60%

Applied Scientist – User Intelligence

$182k – $219k per year · Equity · In office (San Mateo, United States) · Master's Degree
JAX
Multimodal AI
Recommender Systems
PyTorch
Machine Learning
AI Agents
Apply
Top 25% payVerified live · 1 hour ago Visa sponsor 1 month ago Likely open · 64%

Machine Learning Engineer – Ad Sciences

$173k – $210k per year · Middle · 3+ years exp · In office (San Mateo, United States) · Master's Degree · Middle
Java
Python
AI/ML
DeepSpeed
Reinforcement Learning
JAX
Post-training
TPU
LLM
Machine Learning
AI Agents
ONNX
TensorRT
Triton
Recommender Systems
Spark
PyTorch
DevOps
Platform Engineering
Kubernetes
+10
Apply

Techcombank

techcombank.com
techcombank.com · HQ: Hanoi, Vietnam · Financial Services · Commercial & Retail Banks · 1001-5000 employees · Est. 1993 · Trust A · Hiring 53, 6 new
HQ: Hanoi, Vietnam · Financial Services · Commercial & Retail Banks · 1001-5000 employees · Est. 1993 · Trust A · Hiring 53, 6 new
Techcombank (Vietnam Technological and Commercial Joint Stock Bank) is one of the largest commercial banks in Vietnam, offering a comprehensive suite of retail, corporate, and investment banking services. Headquartered in Hanoi, Vietnam, the financial institution serves millions of individual and enterprise clients through an expansive nationwide branch network alongside advanced digital banking platforms. Founded in 1993, it is widely recognized for pioneering customer-centric digital transformation and driving technological innovation across the country's financial sector.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 1 month ago Fading · 30%

Expert, AI Engineering

≈ $19k – $49k per year (Estimated) · Senior · 8+ years exp · In office (Hanoi, Vietnam) · Bachelor's Degree · Senior · English: C1
Scala
Java
SQL
Python
Databases
Apache Kafka
AI/ML
Multi-Agent Systems
RAG
LLM
Machine Learning
AI Agents
LoRA
Fine-tuning
JAX
Knowledge Distillation
NLP
PEFT
LLMOps
DPO
Model Distillation
TensorFlow
Spark
PyTorch
DevOps
GCP
Azure
AWS
Kubernetes
CI/CD
Analytics
A/B Testing
ETL/ELT
Management
Agile
+22
Apply

HTEC

htec.com
htec.com · HQ: Belgrade, Serbia · HQ: Palo Alto, United States · UI/UX Design · IT Outsourcing & Dedicated Teams · IT Consulting & Digital Transformation · Custom Software Development · 51-200 employees · Est. 2008 · Trust B · Hiring 20, 1 new
HQ: Belgrade, Serbia · HQ: Palo Alto, United States · UI/UX Design · IT Outsourcing & Dedicated Teams · IT Consulting & Digital Transformation · Custom Software Development · 51-200 employees · Est. 2008 · Trust B · Hiring 20, 1 new
HTEC is a technology company founded in Belgrade in 2008. It provides product engineering, design and digital consulting for international clients. The company employs thousands of engineers across Europe and the United States.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 4 hours ago 1 month ago Fading · 30%

Principal AI Compiler & Runtime Engineer

$200k – $350k per year · Equity · Lead · Hybrid (Palo Alto, United States) · Bachelor's Degree · Full-Time · Principal
Mojo
C++
Python
C++
TensorFlow C++
LLVM
AI/ML
Quantization
JAX
ONNX
TensorRT
Triton
TPU
ONNX Runtime
MLIR
Apache TVM
XLA
IREE
TensorFlow
PyTorch
Machine Learning
DevOps
Linux
+12
Apply

Baidu

baidu.com
baidu.com · HQ: Beijing, China · Advertising · Foundation Models · Cloud Platforms (IaaS & PaaS) · Autonomous Driving · LLM & Generative AI · Search Engines · 5000+ employees · Est. 2000 · MOEX: BIDU-RM · Trust C · Hiring 1367, 45 new
HQ: Beijing, China · Advertising · Foundation Models · Cloud Platforms (IaaS & PaaS) · Autonomous Driving · LLM & Generative AI · Search Engines · 5000+ employees · Est. 2000 · MOEX: BIDU-RM · Trust C · Hiring 1367, 45 new
Baidu is a Chinese technology company founded in 2000 by Robin Li and Eric Xu that operates the dominant search engine in mainland China. Around search it has built a large advertising business, the Baidu Maps and Baijiahao content platforms, and an AI Cloud division selling compute, data and model services to enterprises and government. The company was one of the first Chinese firms to commit heavily to artificial intelligence, developing the ERNIE family of large language models and the Apollo autonomous driving stack that powers its Apollo Go robotaxi fleet in several Chinese cities.
Eric Xu · Co-Founder & Vice Chairman · decision-maker
Robin Li · Co-Founder, Chairman & Chief Executive Officer · decision-maker
Dou Shen · Executive Vice President, AI Cloud Group · decision-maker
No careers mailbox known
Verified live · 3 hours ago 1 month ago Fading · 31%

AI推理工程师

In office (Shenzhen, China) · Chinese: C1
C++
Python
C++
TensorFlow C++
AI/ML
Triton Inference Server
JAX
TensorRT
CUDA
TPU
NCCL
cuDNN
XLA
TensorFlow
PyTorch
+5
Apply
Likely inactiveVerified live · 3 hours ago 1 year ago Long shot · 3%

语音信号处理算法工程师

Junior · 1+ year exp · In office (Beijing, China) · Junior · Chinese: C1
JAX
TensorFlow
PyTorch
Apply

Aftershoot

aftershoot.com
aftershoot.com · HQ: Delhi, New Delhi, India · Photo & Video Editing Software · Est. 2020 · Startup · Hiring 12
HQ: Delhi, New Delhi, India · Photo & Video Editing Software · Est. 2020 · Startup · Hiring 12
Aftershoot is an AI-powered photo culling and editing software designed to streamline the post-shoot workflow for photographers by automating various tasks to focus on capturing captivating moments.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago 1 month ago Long shot · 21%

Senior Machine Learning Engineer

≈ $60k – $158k per year (Estimated) · Senior · 5+ years exp · Tampere · Remote (Finland) · Master's Degree · Senior
Python
Rust
AI/ML
JAX
Diffusion Models
TensorFlow
Computer Vision
PyTorch
Machine Learning
DevOps
Docker
GCP
Kubernetes
Azure
AWS
+5
Apply

River AI

river.ai
river.ai · HQ: Palo Alto, United States · AI Development Tools · AI Chips & Accelerators · Foundation Models · 201-500 employees · Est. 2026 · Startup · Trust C · Hiring 19, 1 new
HQ: Palo Alto, United States · AI Development Tools · AI Chips & Accelerators · Foundation Models · 201-500 employees · Est. 2026 · Startup · Trust C · Hiring 19, 1 new
River AI is an artificial intelligence company headquartered in Palo Alto, California, and founded in 2026 by xAI co-founder Igor Babuschkin. The company is building what it calls an open AI stack, spanning personal AI models, tooling for developers to train and serve their own models, and a custom system on chip with an onboard machine learning accelerator. It raised 1.1 billion dollars led by General Catalyst with strategic investment from NVIDIA and AMD Ventures.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 4 hours ago Visa sponsorship 1 month ago Long shot · 14%

Software Engineer, Distributed Training

$200k – $420k per year · In office (Palo Alto, United States) · Relocation · Bachelor's Degree
Python
Rust
C++
AI/ML
Fine-tuning
Reinforcement Learning
JAX
Mixture of Experts
PyTorch
LoRA
PEFT
CUDA
Pre-training
NCCL
+5
Apply
Likely inactiveTop 25% pay Visa sponsorship 5 months ago Long shot · 4%

Research Engineer / Research Scientist

$200k – $420k per year · In office (Palo Alto, United States) · Relocation · Bachelor's Degree
Fine-tuning
RLHF
JAX
Multimodal AI
Diffusion Models
PEFT
Transformers
Pre-training
PyTorch
Machine Learning
Reinforcement Learning
Edge AI
+4
Apply
Likely inactiveTop 25% pay Visa sponsorship 5 months ago Long shot · 4%

Software Engineer, River API

$200k – $420k per year · In office (Palo Alto, United States) · Relocation · Bachelor's Degree
Rust
C++
AI/ML
InfiniBand
JAX
Pre-training
PyTorch
Apply

Inductive Bio

inductive.bio
inductive.bio · HQ: New York, United States · Machine Learning · Drug Discovery AI · 51-200 employees · Est. 2021 · Startup · Hiring 5
HQ: New York, United States · Machine Learning · Drug Discovery AI · 51-200 employees · Est. 2021 · Startup · Hiring 5
Inductive Bio builds machine learning models that predict the absorption, distribution, metabolism and toxicity properties of drug candidates. Founded in 2021 in New York, it pools proprietary experimental data across partners to train models no single company could build alone. Medicinal chemists use the predictions to prioritise which molecules to synthesise.
Joshua Haimson · Executive Officer, Director · decision-maker
Benjamin Birnbaum · Executive Officer, Director · decision-maker
Rohan Ganesh · Director
Careers mailbox is known
Verified live · 2 hours ago 1 month ago Fading · 34%

Member of Technical Staff - Machine Learning

$180k – $250k per year · Staff+ · 4+ years exp · Hybrid (New York, United States) · PhD · Full-Time · Staff
Python
AI/ML
Reinforcement Learning
JAX
TensorFlow
PyTorch
Machine Learning
Apply
salute.com · HQ: Franklin, United States · Information Technology · Professional Services · Maintenance, Repair & Operations (MRO) · Data Centers & Colocation · 1001-5000 employees · Trust B · Hiring 108, 19 new
HQ: Franklin, United States · Information Technology · Professional Services · Maintenance, Repair & Operations (MRO) · Data Centers & Colocation · 1001-5000 employees · Trust B · Hiring 108, 19 new
Salute Mission Critical is a data center services company headquartered in Franklin, Wisconsin, offering more than 40 services across the data center lifecycle, from design, build and commissioning to facility operations, remote hands and decommissioning. It works for hyperscale and colocation clients in the US and Europe, operates through its UK unit 2BM, and staffs sites from Rayville, Cheyenne and Sparks to Amsterdam, Dublin and Gavle. It hires critical facility engineers and technicians, data center operations and cleaning technicians, multi-skilled operators, project managers, CMMS planners and recruiters.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 1 month ago Long shot · 22%

AI/ML & Algorithm Engineer

up to $200k per year · Senior · 8+ years exp · In office (United States) · Master's Degree · Full-Time · Staff
Python
AI/ML
LangGraph
AutoGen
Weights & Biases
MLFlow
Fine-tuning
Embeddings
Reinforcement Learning
JAX
Function Calling
CrewAI
Ray
Human-in-the-Loop
Structured Outputs
Semantic Search
Knowledge Graph
LangChain
Prompt Engineering
PyTorch
RAG
LLM
AI Agents
RLHF
Federated Learning
Time Series
Constitutional AI
Machine Learning
Robotics
Reinforcement Learning
Digital Twin
+21
Apply

Syngenta Group

syngenta.com
syngenta.com · HQ: Basel, Switzerland · Agricultural Biotechnology · Seeds · Crop Protection · 5000+ employees · Est. 2000 · Trust B · Hiring 280, 44 new
HQ: Basel, Switzerland · Agricultural Biotechnology · Seeds · Crop Protection · 5000+ employees · Est. 2000 · Trust B · Hiring 280, 44 new
Syngenta is a global agricultural technology and agribusiness enterprise headquartered in Basel, Switzerland. Formed in 2000 through the merger of Novartis Agribusiness and AstraZeneca Agrochemicals, it now operates under the broader Syngenta Group as a subsidiary of Sinochem.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 1 month ago Long shot · 14%

AI Engineering Team Lead

≈ $43k – $108k per year (Estimated) · Lead · 5+ years exp · In office (Beijing, China) · PhD · Full-Time · Senior · English
Python
AI/ML
Multimodal AI
Computer Vision
Machine Learning
JAX
TensorFlow
PyTorch
DevOps
HPC
Apply

Gladstone Institutes

gladstone.org
gladstone.org · San Francisco · Research Institutes · Biotechnology · 501-1000 employees · Est. 2015 · Trust B · Hiring 9
San Francisco · Research Institutes · Biotechnology · 501-1000 employees · Est. 2015 · Trust B · Hiring 9
Our. Gladstone pushes the boundaries of scientific possibility through our five interconnected institutes. With a deep understanding of how the human body works and what goes wrong in disease, we focus on conditions with profound medical, economic, and social impact.
No named decision-maker yet.
No careers mailbox known
Verified live · 7 hours ago 1 month ago Fading · 27%

Research Engineer III (ML)

$88k – $120k per year · Middle · 4+ years exp · In office (San Francisco, United States) · Bachelor's Degree · Full-Time · Middle
OpenCV
Fine-tuning
Reinforcement Learning
JAX
Multimodal AI
Accelerate
Hugging Face
Computer Vision
PyTorch
Machine Learning
DevOps
SLURM
Docker
Kubernetes
+5
Apply
lsu.edu · HQ: Baton Rouge, United States · Higher Education · 1001-5000 employees · Est. 1860 · Trust B · Hiring 198, 16 new
HQ: Baton Rouge, United States · Higher Education · 1001-5000 employees · Est. 1860 · Trust B · Hiring 198, 16 new
Louisiana State University is a public land-grant and sea-grant research university in Baton Rouge, Louisiana, and the flagship of the LSU System, which also includes campuses in Alexandria, Eunice and Shreveport, the LSU AgCenter and Pennington Biomedical Research Center. Founded in 1860 near Pineville as a seminary and military academy, it dedicated its main Baton Rouge campus in 1926 and enrols about 40,000 students. It hires faculty and instructors, research associates, veterinary technicians, custodians and custodial supervisors, equipment operators, buyers, compensation consultants and communications managers.
No named decision-maker yet.
Careers mailbox is known
Verified live · 8 hours ago 1 month ago Long shot · 8%

Research Associate 5

≈ $50k – $105k per year (Estimated) · Junior · 2+ years exp · In office (United States) · Master's Degree · Full-Time · Junior
C++
Python
C++
TensorFlow C++
AI/ML
Multimodal AI
Knowledge Distillation
Multi-Agent Systems
Model Distillation
Machine Learning
AI Agents
JAX
TensorFlow
PyTorch
Robotics
Sensor Fusion
+5
Apply

JPMorganChase

jpmorganchase.com
jpmorganchase.com · HQ: New York, United States · Wealth Management & Financial Advisors · Asset Management & Funds · Investment Banking & M&A Advisory · Commercial & Retail Banks · 5000+ employees · Est. 1799 · Trust A · Hiring 3505, 284 new
HQ: New York, United States · Wealth Management & Financial Advisors · Asset Management & Funds · Investment Banking & M&A Advisory · Commercial & Retail Banks · 5000+ employees · Est. 1799 · Trust A · Hiring 3505, 284 new
JPMorganChase is the largest bank in the United States by assets and one of the most systemically important financial institutions in the world, with a lineage running back through more than a thousand predecessor firms to the 1799 founding of the Bank of the Manhattan Company. It combines a dominant investment bank and markets business with Chase, the largest retail banking franchise in America, plus commercial banking and asset and wealth management. Headquartered in New York, the group is unusual among banks for the scale of its technology spending, running one of the largest engineering organisations of any financial institution and deploying its own internal AI platform across the firm.
Richard Dulik · Executive Officer · decision-maker
Thomas Chase · Executive Officer, Director · decision-maker
Marianne Lake · Chief Executive Officer, Consumer and Community Banking · decision-maker
Jeremy Barnum · Chief Financial Officer · decision-maker
No careers mailbox known
≈ $162k – $417k per year (Estimated) · Head+ · In office (London, United Kingdom) · Master's Degree · Architect
JAX
Multimodal AI
Transformers
Tokenization
Self-Supervised Learning
Time Series
Pre-training
Prompt Engineering
PyTorch
LLM
Machine Learning
Fine-tuning
Transfer Learning
Post-training
+6
Apply
Verified live · 1 hour ago 1 month ago Long shot · 11%

Quantitative Trading & Research - Quantitative Developer - Associate or Vice President

≈ $153k – $358k per year (Estimated) · Head+ · In office (London, United Kingdom) · Bachelor's Degree · Architect
C++
Python
Databases
Apache Kafka
AI/ML
JAX
Ray
CUDA
Spark
PyTorch
DevOps
Linux
Kubernetes
+2
Apply
Verified live · 1 hour ago 1 month ago Long shot · 5%

Quantitative Trading & Research - AI Scientist, Quantitative Research - Associate

≈ $61k – $124k per year (Estimated) · Junior · 2+ years exp · In office (Singapore) · Master's Degree · Junior
JAX
Multimodal AI
Transformers
Tokenization
Self-Supervised Learning
Time Series
Pre-training
Prompt Engineering
PyTorch
LLM
Machine Learning
Fine-tuning
Transfer Learning
Post-training
+6
Apply

Bedrock Robotics

bedrockrobotics.com
bedrockrobotics.com · HQ: San Francisco, United States · Construction Robotics · Robotics · 501-1000 employees · Est. 2024 · Startup · Trust A · Hiring 50, 4 new
HQ: San Francisco, United States · Construction Robotics · Robotics · 501-1000 employees · Est. 2024 · Startup · Trust A · Hiring 50, 4 new
Bedrock Robotics is a San Francisco company developing autonomous driving systems that retrofit heavy construction equipment such as excavators, helping contractors add capacity and speed up infrastructure and site work. Founded in 2024 by veterans of self-driving and robotics programs, it has engineering teams in San Francisco and New York and field operations and testing in Lockhart, Texas. Its openings are largely engineering roles in state estimation and localization, behavior and world-model machine learning, onboard inference infrastructure, sensor systems, fleet platform software, hardware integration and validation, including a large 2027 internship program.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago Visa sponsor 1 month ago Likely open · 48%

2027 Internship Onboard Infrastructure Engineer, ML Inference

Intern · In office (San Francisco, United States) · Bachelor's Degree · Internship · Intern
Rust
C++
AI/ML
vLLM
JAX
Multimodal AI
TensorRT
Tokenization
CUDA
Edge AI
ExecuTorch
Vision-Language-Action
PyTorch
LLM
Quantization
ONNX Runtime
Speculative Decoding
KV Cache
DevOps
eBPF
+10
Apply
Open 177 daysVerified live · 3 hours ago Visa sponsor 5 months ago Long shot · 18%

Machine Learning Engineer: Perception

≈ $168k – $323k per year (Estimated) · Senior · 5+ years exp · Hybrid (San Francisco, United States) · Full-Time · Senior
Rust
C++
Python
C++
TensorFlow C++
AI/ML
JAX
TensorFlow
Computer Vision
PyTorch
Machine Learning
+1
Apply
Open 271 daysVerified live · 3 hours ago Visa sponsor 9 months ago Long shot · 12%

Machine Learning Engineer: Imitation and Reinforcement Learning for Robotics

≈ $140k – $268k per year (Estimated) · Middle · 3+ years exp · Hybrid (San Francisco, United States) · Full-Time · Middle
Reinforcement Learning
JAX
Vision-Language-Action
TensorFlow
PyTorch
Machine Learning
Robotics
Reinforcement Learning
Apply

Bosch

bosch.com
bosch.com · HQ: Gerlingen, Germany · HVAC & Refrigeration · Industrial Automation · Sensors & Transducers · Motors, Drives & Motion Control · Home Appliances · Tools & DIY Supplies · Automotive Components · 5000+ employees · Est. 1886 · Trust B · Hiring 3653, 328 new
HQ: Gerlingen, Germany · HVAC & Refrigeration · Industrial Automation · Sensors & Transducers · Motors, Drives & Motion Control · Home Appliances · Tools & DIY Supplies · Automotive Components · 5000+ employees · Est. 1886 · Trust B · Hiring 3653, 328 new
Bosch is a German engineering and technology group founded in Stuttgart in 1886 by Robert Bosch as a workshop for precision mechanics and electrical engineering. It is the largest automotive supplier in the world, making braking, steering, powertrain, sensor and driver assistance systems for nearly every carmaker, and it also produces power tools, home appliances through BSH, industrial drives through Rexroth, and building security and heating systems. Around ninety per cent of the company is owned by a charitable foundation, an ownership structure that gives it unusual freedom to invest across long research cycles including semiconductors and hydrogen.
Markus Forschner · Chief Financial Officer · decision-maker
Stefan Hartung · Chairman of the Board of Management · decision-maker
Robert Bosch · Founder · founder
Careers mailbox is known
Verified live · 1 hour ago 1 month ago Long shot · 18%

AI Research Scientist - Safety

$165k – $185k per year · Middle · 3+ years exp · In office (Sunnyvale, United States) · PhD · Full-Time · Middle · English
Python
C++
AI/ML
Fine-tuning
JAX
Self-Supervised Learning
Time Series
Explainable AI
World Models
Computer Vision
PyTorch
Machine Learning
Interpretability
+4
Apply

Inferact

inferact.ai
inferact.ai · HQ: San Francisco, United States · AI Development Tools · AI Compute & Inference · 51-200 employees · Startup · Trust A · Hiring 32
HQ: San Francisco, United States · AI Development Tools · AI Compute & Inference · 51-200 employees · Startup · Trust A · Hiring 32
Inferact is a startup founded by creators and core maintainers of vLLM, the most popular open-source LLM inference engine. Our mission is to grow vLLM as the world.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago Visa sponsor 1 month ago Long shot · 17%

Inference Engineering, Co-op

≈ $225k – $500k per year (Estimated) · In office (San Francisco, United States) · Bachelor's Degree · Internship
Rust
Go
C++
Python
AI/ML
vLLM
Quantization
JAX
Multimodal AI
Mixture of Experts
CUDA
Triton
TPU
NCCL
InfiniBand
ROCm
MLIR
XLA
CUTLASS
KV Cache
SGLang
TensorRT
TensorRT-LLM
Ray
PyTorch
DevOps
Kubernetes
Helm
SLURM
Terraform
+20
Apply
Open 125 daysTop 25% pay Visa sponsorship 4 months ago Long shot · 9%

Member of Technical Staff, TPU Performance Engineering

$156k – $313k per year · Staff+ · In office (Singapore) · Full-Time · Staff
vLLM
JAX
TPU
XLA
KV Cache
Machine Learning
Quantization
SGLang
TensorRT
TensorRT-LLM
MLIR
PyTorch
LLM
+5
Apply

Epsilon Health

epsilon.health
epsilon.health · HQ: San Francisco, United States · Medical AI · Radiology & Imaging Centers · 51-200 employees · Startup · Hiring 7
HQ: San Francisco, United States · Medical AI · Radiology & Imaging Centers · 51-200 employees · Startup · Hiring 7
Epsilon Health is an AI-native radiology practice that reinvents how medical imaging is interpreted, built on a large and diverse medical imaging dataset. It hires research scientists for vision foundation models and post-training and software engineers for machine learning infrastructure. The company is headquartered in San Francisco, California, United States.
No named decision-maker yet.
Careers mailbox is known
Verified live · 4 hours ago 1 month ago Fading · 34%

Software Engineer - ML Infrastructure

≈ $175k – $338k per year (Estimated) · Senior · 6+ years exp · In office (San Francisco, United States) · Full-Time · Senior
Python
AI/ML
DeepSpeed
Reinforcement Learning
JAX
Multimodal AI
Post-training
Megatron-LM
FSDP
PyTorch
Machine Learning
vLLM
Quantization
SGLang
TensorRT
Triton
Prompt Caching
DevOps
Kubernetes
Analytics
A/B Testing
+10
Apply
Verified live · 4 hours ago 2 months ago Long shot · 19%

Research Engineer / Research Scientist, Vision-Language Model Pretraining

≈ $170k – $377k per year (Estimated) · In office (San Francisco, United States) · Full-Time
Reinforcement Learning
JAX
Multimodal AI
VLM
Post-training
PyTorch
Machine Learning
Qwen
Contrastive Learning
LLaVA
Interpretability
Computer Vision
+4
Apply
Verified live · 4 hours ago 2 months ago Long shot · 19%

Research Engineer / Research Scientist, Vision Foundation Models

≈ $163k – $364k per year (Estimated) · In office (San Francisco, United States) · Full-Time
JAX
Multimodal AI
Transfer Learning
Self-Supervised Learning
Contrastive Learning
Computer Vision
PyTorch
DINOv2
Interpretability
+1
Apply

True Anomaly

trueanomaly.space
trueanomaly.space · HQ: Centennial, United States · Satellite Manufacturing · Space Defense · Space Services · 1001-5000 employees · Est. 2022 · Trust A · Hiring 173, 5 new
HQ: Centennial, United States · Satellite Manufacturing · Space Defense · Space Services · 1001-5000 employees · Est. 2022 · Trust A · Hiring 173, 5 new
True Anomaly is an American space defence company founded in 2022 by former United States Space Force officers to build spacecraft and software for orbital security missions. Its Jackal autonomous orbital vehicle is designed for rendezvous, proximity operations and inspection of other satellites, and its Mosaic software provides mission planning, simulation and command and control for space operations, sold mainly to the Space Force and allied defence customers. Headquartered in Centennial, Colorado, the company manufactures its own spacecraft and has raised substantial venture funding to scale production.
Even Rogers · Co-founder and Chief Executive Officer · decision-maker
Thomas Nichols · Executive Officer, Director · decision-maker
Tom Nichols · Co-founder · founder
Daniel Brunski · Director
Careers mailbox is known
Top 25% payVerified live · 4 hours ago 1 month ago Likely open · 64%

Machine Learning Engineer (II-III), Space Edge Deployment

$125k – $220k per year · Middle · 2+ years exp · In office (Denver, Long Beach, United States) · Bachelor's Degree · Contractor · Nationals only · Citizenship or vetting required · Middle
Python
AI/ML
JAX
Anomaly Detection
TensorFlow
PyTorch
Machine Learning
Time Series
Computer Vision
DevOps
Git
+1
Apply
Top 25% payVerified live · 4 hours ago 5 months ago Fading · 25%

Machine Learning Engineer (II-III), Space Agentic AI

$125k – $220k per year · Middle · 2+ years exp · In office (Denver, Long Beach, United States) · Bachelor's Degree · Contractor · Nationals only · Citizenship or vetting required · Middle
Python
AI/ML
JAX
Anomaly Detection
TensorFlow
PyTorch
Machine Learning
Time Series
Computer Vision
DevOps
Git
+1
Apply

V2X

gov2x.com
gov2x.com · HQ: Reston, United States · Artificial Intelligence · Military · Autonomous Driving · 1001-5000 employees · Trust B · Hiring 472, 51 new
HQ: Reston, United States · Artificial Intelligence · Military · Autonomous Driving · 1001-5000 employees · Trust B · Hiring 472, 51 new
Headquartered in Reston, Virginia, V2X is a leading global defense contractor and government services provider that delivers critical mission-readiness solutions across defense, national security, and civilian agencies. Formed through the merger of Vectrus and Vertex Aerospace, the company specializes in aerospace maintenance and modernization, logistics, base operations support, and mission-focused IT and cybersecurity solutions. Its multidisciplinary teams provide comprehensive infrastructure sustainment and operational capabilities across all phases of the mission lifecycle in military theaters and federal facilities worldwide.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 1 month ago Long shot · 17%

Agentic AI Engineer

≈ $94k – $212k per year (Estimated) · Junior · 2+ years exp · In office (Indianapolis, United States) · Bachelor's Degree · Secret · Citizenship or vetting required · Junior
Python
AI/ML
JAX
Agentic Workflows
TensorFlow
LangChain
PyTorch
Machine Learning
AI Agents
Reinforcement Learning
Explainable AI
Synthetic Data
Multi-Agent Systems
Robotics
Sensor Fusion
Reinforcement Learning
+6
Apply

SAIGroup

saigroup.ai
saigroup.ai · Bengaluru · Los Altos · Palo Alto · Investment Holding Companies · Private Equity · 5000+ employees · Trust A · Hiring 13, 4 new
Bengaluru · Los Altos · Palo Alto · Investment Holding Companies · Private Equity · 5000+ employees · Trust A · Hiring 13, 4 new
Growing companies with advanced AI. SAIGroup is a private investment firm investing in businesses with the potential to become leaders in enterprise AI solutions. SAIGroup's investments enable our portfolio companies to accelerate their innovation and growth. Principal owner and investor Dr.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 1 month ago Fading · 31%

Senior Agentic AI Engineer — Agentic Medical AI

≈ $24k – $50k per year (Estimated) · Senior · In office (Bengaluru, India, Palo Alto, United States) · Senior
Embeddings
JAX
Multimodal AI
Chain-of-Thought
NLP
VLM
CLIP
Contrastive Learning
Human-in-the-Loop
Interpretability
PyTorch
LLM
AI Agents
LlamaIndex
Haystack
LangChain
MedTech
DICOM
+9
Apply
Machines can read this selection · checked 20:38 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.