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

Citi

citi.com
citi.com • HQ: New York, United States • Cards & Card Issuing • Payment Processing & Gateways • Wealth Management & Financial Advisors • Investment Banking & M&A Advisory • Commercial & Retail Banks • 5000+ employees • Est. 1812 • Trust A • Hiring 2594, 93 new
HQ: New York, United States • Cards & Card Issuing • Payment Processing & Gateways • Wealth Management & Financial Advisors • Investment Banking & M&A Advisory • Commercial & Retail Banks • 5000+ employees • Est. 1812 • Trust A • Hiring 2594, 93 new
Citi is one of the largest banks in the world, tracing its lineage to the City Bank of New York founded in 1812 and taking its modern shape through the 1998 merger that created Citigroup. Its most distinctive asset is a cross-border payments and treasury network unmatched by any competitor, moving trillions of dollars a day for multinational corporations, governments and other banks across roughly ninety countries. Alongside that institutional franchise it runs markets and investment banking, wealth management and a United States personal bank, and has spent recent years simplifying itself by exiting consumer operations across Asia, Europe and Latin America.
Vis Raghavan · Head of Banking · decision-maker
Mark Mason · Chief Financial Officer · decision-maker
Jane Fraser · Chief Executive Officer · decision-maker
Sanford Weill · Architect of the 1998 Merger
Careers mailbox is known
Verified live · 4 hours ago Visa sponsor 11 days ago Likely open · 47%

Python AI Engineer - Assistant Vice President

$107k – $161k per year • Head+ • 5+ years exp • Hybrid (Irving, United States) • Bachelor's Degree • Full-Time • Architect
Python
AI/ML
JAX
Diffusion Models
TensorFlow
PyTorch
Google ADK
Machine Learning
DevOps
GCP
Azure
AWS
Docker
Kubernetes
+4
Apply
Verified live · 4 hours ago 1 month ago Long shot · 14%

Full Stack Engineer - Senior Vice President

≈ $74k – $147k per year (Estimated) • Head+ • 10+ years exp • Hybrid (Mississauga, Canada) • Bachelor's Degree • Full-Time • Senior
Java
Java
Maven
Spring Boot
Gradle
Databases
Apache Kafka
AI/ML
JAX
DevOps
Rest API
WebSockets
Platform Engineering
SOAP
Apache HTTP Server
+3
Apply
Verified live · 4 hours ago 2 months ago Long shot · 10%

JAVA Microservices Engineering Lead

$85k – $121k per year • Lead • 6+ years exp • In office (Mississauga, Canada) • Bachelor's Degree • Full-Time • Staff
JavaScript
Java
TypeScript
COBOL
Java
Spring Boot
Spring Security
Spring Cloud
Hazelcast
COBOL
IBM MQ
Databases
Redis
RabbitMQ
Memcached
Apache Kafka
AI/ML
JAX
Frontend
Angular
DevOps
GCP
OpenShift
Azure
CI/CD
Jenkins
Git
AWS
Kubernetes
Gitflow
Configuration Management
Tekton
SOAP
Dynatrace
AppDynamics
API Gateway
Linux
+23
Apply

Lyft

lyft.com
lyft.com • HQ: San Francisco, United States • Shared Mobility • Ride Hailing • 1001-5000 employees • Est. 2012 • NASDAQ: LYFT • Trust B • Hiring 180, 13 new
HQ: San Francisco, United States • Shared Mobility • Ride Hailing • 1001-5000 employees • Est. 2012 • NASDAQ: LYFT • Trust B • Hiring 180, 13 new
Lyft is an American ride-hailing company that connects riders with drivers through its mobile app in the United States and Canada, and also runs bike and scooter sharing systems in several cities. Its business includes Lyft Business programmes for companies, the Flexdrive vehicle rental service for drivers and Lyft Urban Solutions for municipal bikeshare operations. The company is listed on Nasdaq and employs product, engineering and operations staff in San Francisco, Toronto, New York and Montreal. 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 11 days ago Likely open · 61%

PhD Machine Learning Software Engineer Intern (Summer 2027)

$130k – $136k per year • Intern • Hybrid (San Francisco, United States) • PhD • Internship • Intern
Python
SQL
AI/ML
Reinforcement Learning
JAX
TensorFlow
PyTorch
Machine Learning
Spark
Recommender Systems
Analytics
A/B Testing
+2
Apply
pdw.ai • HQ: Huntsville, United States • Military Drones • Drones & UAVs • 201-500 employees • Est. 2018 • Trust A • Hiring 27, 2 new
HQ: Huntsville, United States • Military Drones • Drones & UAVs • 201-500 employees • Est. 2018 • Trust A • Hiring 27, 2 new
Performance Drone Works builds rugged small unmanned aircraft for United States special operations. Its C100 drone carries payloads in contested electronic environments. The company designs and manufactures in Alabama.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago Visa sponsor 11 days ago Hiring now · 90%

Senior AI/ML Engineer

≈ $136k – $259k per year (Estimated) • Senior • 7+ years exp • In office (Denver, United States) • Bachelor's Degree • Full-Time • Senior
Python
AI/ML
Fine-tuning
Embeddings
Quantization
JAX
Prompt Engineering
Function Calling
AI Agents
TensorFlow
PyTorch
RAG
Reranking
LLM Guardrails
Agentic Workflows
Machine Learning
DevOps
CI/CD
+8
Apply

Illumina

illumina.com
illumina.com • HQ: San Diego, United States • Laboratory Equipment • Genomics & Genetic Testing • 1001-5000 employees • Est. 1998 • NASDAQ: ILMN • Trust B • Hiring 145, 17 new
HQ: San Diego, United States • Laboratory Equipment • Genomics & Genetic Testing • 1001-5000 employees • Est. 1998 • NASDAQ: ILMN • Trust B • Hiring 145, 17 new
Illumina is an American company founded in 1998 whose sequencing machines drove the cost of reading a human genome from hundreds of millions of dollars to a few hundred, and it holds the dominant share of the market it largely created. Its business is a classic instruments and consumables model: sequencers are sold once and then consume proprietary reagent kits for years, which is where most of the profit sits. Headquartered in San Diego and listed on Nasdaq, it spent years entangled in the forced divestiture of the cancer test maker Grail and now faces genuine competition from Chinese and startup sequencing platforms for the first time.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 1 day ago 11 days ago Likely open · 63%

Senior Bioinformatics Scientist – AI/ML for Protein Engineering

$129k – $194k per year • Senior • 5+ years exp • In office (San Diego, United States) • Bachelor's Degree • Full-Time • Senior
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
Machine Learning
JAX
AI Agents
TensorFlow
PyTorch
DevOps
Git
Unix
+3
Apply
Verified live · 1 day ago 15 days ago Likely open · 55%

Senior Staff Bioinformatics Scientist (AI-guided protein engineering)

≈ $58k – $138k per year (Estimated) • Staff+ • In office (Singapore) • PhD • Full-Time • Staff
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
JAX
AI Agents
TensorFlow
PyTorch
Agentic Workflows
DevOps
Git
Unix
+3
Apply

Mistral AI

mistral.ai
mistral.ai • HQ: Paris, France • Code Intelligence • AI Research Labs • LLM & Generative AI • Foundation Models • 1001-5000 employees • Est. 2023 • Trust A • Hiring 216, 12 new
HQ: Paris, France • Code Intelligence • AI Research Labs • LLM & Generative AI • Foundation Models • 1001-5000 employees • Est. 2023 • Trust A • Hiring 216, 12 new
Mistral AI is a French artificial intelligence company founded in Paris in 2023 by former researchers from Google DeepMind and Meta. It builds efficient open-weight and commercial large language models, including the Mistral, Mixtral, Magistral and Codestral families, and distributes them through its own developer platform as well as the major cloud marketplaces. Positioned as Europe's leading frontier model developer, the company also ships the Le Chat assistant, on-premise deployment options for regulated industries and a sovereign compute offering for customers who must keep data inside the region.
Timothee Lacroix · Co-Founder & Chief Technology Officer · decision-maker
Guillaume Lample · Co-Founder & Chief Scientist · decision-maker
Arthur Mensch · Co-Founder & Chief Executive Officer · decision-maker
Timothée Lacroix · Directeur Général · decision-maker
Careers mailbox is known
Verified live · 1 hour ago 11 days ago Hiring now · 87%

LLM Scientist, AI for Science

≈ $58k – $124k per year (Estimated) • Hybrid (Paris, France) • Relocation • Full-Time
Fine-tuning
RLHF
Reinforcement Learning
JAX
AI Agents
Mistral
TensorFlow
PyTorch
LLM
Synthetic Data
Post-training
Edge AI
DevOps
HPC
+5
Apply
Verified live · 1 hour ago 22 days ago Hiring now · 80%

Applied Science - Research Engineer (Robotics)

≈ $40k – $109k per year (Estimated) • In office (Singapore) • Relocation • Master's Degree • Full-Time • English: C1
Python
AI/ML
JAX
VLM
Mistral SDK
PyTorch
Post-training
Pre-training
Vision-Language-Action
Machine Learning
Robotics
Isaac Sim
+2
Apply
Verified live · 1 hour ago Visa sponsorship 29 days ago Likely open · 60%

Applied Scientist (Internship in Paris or London)

Intern • In office (Seoul, South Korea) • Relocation • Master's Degree • Internship • Intern • English: C1
Python
AI/ML
JAX
Mistral SDK
PyTorch
Post-training
Pre-training
Multimodal AI
Diffusion Models
Time Series Forecasting
World Models
Machine Learning
+3
Apply

EPAM Systems Poland

epam-poland.pl
epam-poland.pl • HQ: Kraków, Poland • Custom Software Development • IT Consulting & Digital Transformation • 1001-5000 employees • Est. 2011 • Hiring 400, 27 new
HQ: Kraków, Poland • Custom Software Development • IT Consulting & Digital Transformation • 1001-5000 employees • Est. 2011 • Hiring 400, 27 new
Founded in 2011, EPAM Systems Poland is a major software engineering and IT consulting firm specializing in custom digital platforms, cloud architecture, and data services. Headquartered in Krakow, Poland, the company operates development centers across several major cities to serve leading global enterprises. Through its extensive network of tech talent and digital innovation hubs, it plays a pivotal role in driving regional technology solutions and digital transformation.
No named decision-maker yet.
No careers mailbox known
≈ $75k – $120k per year (Estimated) • Staff+ • 12+ years exp • Hybrid (Kraków, Poland) • Full-Time • Staff
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
JAX
TensorFlow
PyTorch
TPU
MLIR
Machine Learning
Anthropic
DevOps
GCP
Azure
AWS
Platform Engineering
Management
Agile
+8
Apply

Aisensy

aisensy.com
aisensy.com • HQ: Bengaluru, India • Messaging Apps • Marketing & Sales AI • Est. 2020 • Likely startup • Hiring 3
HQ: Bengaluru, India • Messaging Apps • Marketing & Sales AI • Est. 2020 • Likely startup • Hiring 3
AiSensy is the AI-first WhatsApp Marketing Platform built on official WhatsApp Business APIs. Broadcast offers, automate notifications & solve for support.
No named decision-maker yet.
Careers mailbox is known
11 days ago Hiring now · 70%

Applied Researcher

≈ $14k – $34k per year (Estimated) • Junior • 2+ years exp • In office (Bengaluru, India) • Junior
Python
AI/ML
Fine-tuning
RLHF
JAX
PyTorch
LLM
Synthetic Data
Post-training
Machine Learning
DevOps
Docker
Kubernetes
+3
Apply

ZenteiQ

zenteiq.ai
zenteiq.ai • HQ: Bengaluru, India • STEM Education • Machine Learning • Simulation & Digital Twin Software • AI for Science • Est. 2022 • Likely startup • Hiring 10, 1 new
HQ: Bengaluru, India • STEM Education • Machine Learning • Simulation & Digital Twin Software • AI for Science • Est. 2022 • Likely startup • Hiring 10, 1 new
ZenteiQ is a deep technology company headquartered in Bengaluru, India, and founded in 2022 out of research at the Indian Institute of Science. The company works on scientific machine learning, combining numerical simulation with neural networks for engineering and industrial modelling problems, and builds training programmes around those methods. It sells to industrial and research customers in India while running education initiatives that push scientific computing skills into the wider engineering workforce.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago 11 days ago Hiring now · 72%

AI Research Engineer

≈ $24k – $55k per year (Estimated) • Middle • 3+ years exp • In office (Bengaluru, India) • Full-Time • Middle
DeepSpeed
vLLM
Fine-tuning
Reinforcement Learning
JAX
Chain-of-Thought
PyTorch
LLM
Ray
DPO
PPO
GRPO
Post-training
Constitutional AI
RLAIF
Reward Modeling
+8
Apply
Verified live · 1 hour ago 1 month ago Long shot · 22%

ML Systems Engineer – Training & Inference

In office (Bengaluru, India) • Full-Time
Python
Java
Rust
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
TPU
XLA
Weights & Biases
MLFlow
Vertex AI
JAX
TensorFlow
PyTorch
LLM
Ray
DevOps
CI/CD
Kubernetes
Google GKE
Terraform
GCP
+13
Apply
Verified live · 1 hour ago 1 month ago Long shot · 19%

Distributed ML/AI Infrastructure Engineer

≈ $23k – $53k per year (Estimated) • Middle • 3+ years exp • In office (Bengaluru, India) • Full-Time • Middle
Python
AI/ML
CUDA Toolkit
PyTorch
CUDA
TPU
NCCL
JAX
LLM
TensorBoard
Ray
Triton
InfiniBand
XLA
DevOps
Git
Docker
Kubernetes
HPC
Linux
SLURM
+11
Apply

Western Digital

wdc.com
wdc.com • HQ: San Jose, United States • Digital Storage • Enterprise Storage Systems • 5000+ employees • Est. 1970 • NASDAQ: WDC • Trust B • Hiring 246, 22 new
HQ: San Jose, United States • Digital Storage • Enterprise Storage Systems • 5000+ employees • Est. 1970 • NASDAQ: WDC • Trust B • Hiring 246, 22 new
Western Digital is an American data storage company founded in 1970 that, since separating its flash memory business as SanDisk in early 2025, concentrates entirely on hard disk drives. Its drives are sold overwhelmingly to cloud and enterprise data centre operators rather than to consumers, and the company competes with Seagate in a market where areal density technologies such as shingled and heat-assisted recording determine cost per terabyte. Headquartered in San Jose and listed on Nasdaq, it has benefited from the sharp increase in storage demand created by artificial intelligence workloads.
Graham Leslie Holmes · Geschäftsführer · decision-maker
Irving Tan · Chief Executive Officer · decision-maker
Irwin Federman · Former Chairman
No careers mailbox known
Verified live · 3 hours ago 11 days ago Likely open · 60%

AI Research Scientist, Scientific ML

≈ $47k – $96k per year (Estimated) • Junior • 1+ year exp • In office (Singapore) • Master's Degree • Full-Time • Senior • English
JAX
Diffusion Models
PyTorch
Synthetic Data
Machine Learning
Fine-tuning
GNN
Robotics
Digital Twin
Apply

Flagship Pioneering

flagshippioneering.com
flagshippioneering.com • HQ: Cambridge, United States • Venture Capital • 1001-5000 employees • Est. 2000 • Trust A • Hiring 93, 3 new
HQ: Cambridge, United States • Venture Capital • 1001-5000 employees • Est. 2000 • Trust A • Hiring 93, 3 new
Flagship Pioneering is a life sciences venture creation and platform investment firm headquartered in Cambridge, Massachusetts. Founded in 2000 by CEO Noubar Afeyan, the firm operates on a novel institutional innovation model where it acts as a principal founder, funder, and operator of first-in-category biotechnology and sustainability companies.
No named decision-maker yet.
Careers mailbox is known
Verified live · 15 min ago Visa sponsor 12 days ago Hiring now · 90%

Senior Machine Learning Scientist

≈ $110k – $205k per year (Estimated) • Senior • Hybrid (Cambridge, United Kingdom) • PhD • Senior
Python
AI/ML
JAX
AI Agents
PyTorch
LLM
Agentic Workflows
Machine Learning
Cybersecurity
GDPR
Apply
Top 25% payVerified live · 15 min ago Visa sponsor 22 days ago Hiring now · 89%

Senior Director / Vice President, Machine Learning (Generative Chemistry)

$279k – $413k per year • Head+ • 7+ years exp • In office (Cambridge, United States) • Master's Degree • Senior
Python
Python
FastAPI
Databases
PostgreSQL
Snowflake
pgvector
FAISS
Amazon Redshift
AI/ML
JAX
Multimodal AI
AI Agents
TensorFlow
PyTorch
LLM
Agentic Workflows
MLFlow
Prefect
Gradio
Streamlit
Amazon SageMaker
DevOps
Terraform
GitHub Actions
CI/CD
AWS
Docker
Kubernetes
Amazon EKS
GitHub
Amazon S3
Amazon ECS
AWS Step Functions
Cybersecurity
GDPR
+23
Apply
Top 25% payVerified live · 15 min ago Visa sponsor 2 months ago Fading · 34%

Principal Scientist, Machine Learning

$208k – $286k per year • Lead • 5+ years exp • In office (Cambridge, United States) • Master's Degree • Principal
Python
AI/ML
AI Agents
TensorFlow
LLM
Agentic Workflows
Machine Learning
LangGraph
AutoGen
LangChain
Vertex AI
JAX
Pydantic AI
PyTorch
CrewAI
OpenAI
Anthropic
DevOps
FinOps
Cybersecurity
GDPR
+10
Apply

Sciforium

sciforium.ai
sciforium.ai • HQ: San Francisco, United States • AI Compute & Inference • 201-500 employees • Est. 2024 • Startup • Trust A • Hiring 13
HQ: San Francisco, United States • AI Compute & Inference • 201-500 employees • Est. 2024 • Startup • Trust A • Hiring 13
Access state-of-the-art AI models through a simple, fast, and reliable API. Deploy production-ready AI applications in minutes, not months.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 2 hours ago Visa sponsor 12 days ago Fading · 43%

Senior HPC & GPU Infrastructure Engineer

$150k – $220k per year • Senior • 5+ years exp • In office (San Francisco, United States) • Bachelor's Degree • Full-Time • Senior
Python
Bash
AI/ML
vLLM
CUDA Toolkit
JAX
Multimodal AI
PyTorch
CUDA
NCCL
NVLink
ROCm
cuDNN
Machine Learning
DevOps
Ubuntu
CentOS Stream
HPC
Linux
Terraform
Ansible
SLURM
Kubernetes
SaltStack
Configuration Management
Cybersecurity
LDAP
+16
Apply
Top 25% payVerified live · 2 hours ago Visa sponsor 1 month ago Long shot · 21%

GPU Kernel Engineer

$190k – $250k per year • Senior • 5+ years exp • Hybrid (San Francisco, United States) • Bachelor's Degree • Full-Time • Senior
Python
C++
Assembly
C++
PyTorch C++
AI/ML
CUDA Toolkit
JAX
Multimodal AI
PyTorch
LLM
CUDA
Triton
ROCm
Edge AI
vLLM
TensorRT
TPU
XLA
+9
Apply
Top 25% payVerified live · 2 hours ago Visa sponsor 2 months ago Long shot · 16%

Pre-training Research Engineer

$165k – $225k per year • Senior • 5+ years exp • In office (San Francisco, United States) • Master's Degree • Full-Time • Senior
Multimodal AI
Pre-training
Machine Learning
JAX
Flax
Megatron-LM
FSDP
XLA
Apply
utexas.edu • HQ: Austin, United States • Higher Education • 1001-5000 employees • Est. 1883 • Trust B • Hiring 491, 38 new
HQ: Austin, United States • Higher Education • 1001-5000 employees • Est. 1883 • Trust B • Hiring 491, 38 new
The University of Texas at Austin is the flagship campus of the University of Texas System, opened in 1883 and now one of the largest public research universities in the United States. It enrols around fifty thousand students across colleges covering engineering, natural sciences, business, liberal arts, law, computer science and communications, and it operates major research units including the Texas Advanced Computing Center and the Oden Institute for computational engineering. Located in Austin, it is funded partly through the Permanent University Fund built on oil and gas revenue from state land, and it employs tens of thousands of academic and support staff.
No named decision-maker yet.
Careers mailbox is known
Verified live · 9 hours ago Visa sponsor 12 days ago Likely open · 55%

AI Computational Scientist – Senior Software Engineer

≈ $109k – $207k per year (Estimated) • Senior • 3+ years exp • In office (United States) • Master's Degree • Full-Time • Senior
Python
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
DevOps
Linux
Apply
ifm.ai • HQ: Sunnyvale, United States • AI Research Labs • Foundation Models • Likely startup • Trust A • Hiring 44
HQ: Sunnyvale, United States • AI Research Labs • Foundation Models • Likely startup • Trust A • Hiring 44
The Institute of Foundation Models (IFM) is an AI research lab developing open frontier foundation models.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 1 hour ago 12 days ago Hiring now · 90%

Machine Learning Engineer — Reinforcement Learning

$150k – $450k per year • Senior • 5+ years exp • In office (Sunnyvale, United States) • Full-Time • Senior
Python
AI/ML
DeepSpeed
vLLM
JAX
SGLang
PyTorch
LLM
Ray
PPO
GRPO
FSDP
NCCL
Machine Learning
CUDA Toolkit
CUDA
Triton
Pre-training
DevOps
SLURM
Kubernetes
+11
Apply
Top 25% payVerified live · 1 hour ago 12 days ago Hiring now · 90%

Machine Learning Engineer — Pre-training (LLM)

$150k – $450k per year • Senior • 5+ years exp • In office (Sunnyvale, United States) • Full-Time • Senior
Python
AI/ML
DeepSpeed
JAX
PyTorch
Ray
Pre-training
FSDP
NCCL
Machine Learning
CUDA Toolkit
CUDA
Triton
DevOps
SLURM
Kubernetes
+6
Apply
Verified live · 1 hour ago 1 year ago Fading · 39%

High Performance Computing Software Engineer - Supercomputing

In office (Abu Dhabi, United Arab Emirates) • Full-Time
DeepSpeed
JAX
TensorFlow
PyTorch
NCCL
Machine Learning
DevOps
SLURM
Kubernetes
HPC
Linux
+2
Apply

Humanoid

thehumanoid.ai
thehumanoid.ai • HQ: London, United Kingdom • Industrial Robotics • Logistics & Warehouse Robotics • Robotics AI • Humanoid Robots • 201-500 employees • Est. 2024 • Startup • Trust B • Hiring 99, 10 new
HQ: London, United Kingdom • Industrial Robotics • Logistics & Warehouse Robotics • Robotics AI • Humanoid Robots • 201-500 employees • Est. 2024 • Startup • Trust B • Hiring 99, 10 new
Humanoid is a British robotics company founded in 2024 that is developing general-purpose humanoid robots for industrial and logistics work. Its argument for a human form is practical rather than aesthetic: warehouses, factories and buildings were designed around human reach, stride and grip, so a robot shaped like a person can be deployed without rebuilding the environment around it. Headquartered in London and backed by venture investors, it is developing both a wheeled and a bipedal platform and competes with considerably better-funded American and Chinese efforts in the same field.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 12 days ago Likely open · 62%

Autonomy Engineer - VLA Pre-training

≈ $145k – $273k per year (Estimated) • Middle • 3+ years exp • In office (San Diego, United States) • Full-Time • Middle
Python
AI/ML
JAX
PyTorch
Synthetic Data
Post-training
Pre-training
Edge AI
Vision-Language-Action
Robotics
OpenVLA
+1
Apply
Verified live · 3 hours ago 3 months ago Long shot · 13%

Deep Learning Engineer - World Models

Equity • In office (London, United Kingdom) • Full-Time
Python
AI/ML
JAX
Multimodal AI
Diffusion Models
PyTorch
Vision-Language-Action
World Models
Synthetic Data
Robotics
Teleoperation
Sim-to-Real
Isaac Sim
MuJoCo
+4
Apply
Verified live · 3 hours ago 3 months ago Long shot · 21%

Reinforcement Learning Engineer - Manipulation

≈ $80k – $180k per year (Estimated) • Equity • Middle • 3+ years exp • In office (London, United Kingdom) • Full-Time • Middle
Python
AI/ML
Reinforcement Learning
JAX
PyTorch
Ray
Robotics
Reinforcement Learning
Isaac Sim
MuJoCo
OpenVLA
+1
Apply

Atoms

atoms.co
atoms.co • HQ: Los Angeles, United States • Mining & Heavy Industry Robotics • Industrial Automation • Industrial Robotics • 1001-5000 employees • Trust A • Hiring 296, 27 new
HQ: Los Angeles, United States • Mining & Heavy Industry Robotics • Industrial Automation • Industrial Robotics • 1001-5000 employees • Trust A • Hiring 296, 27 new
Atoms is an industrial robotics company that rebranded from City Storage Systems, the parent of CloudKitchens. It builds task-specific robots and automation infrastructure across food, mining and transport, and its openings are concentrated in San Francisco, Los Angeles, Pittsburgh and New York. The company is based in the United States.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 1 hour ago 12 days ago Hiring now · 87%

Machine Learning Engineer Intern

$140k per year • Intern • In office (San Francisco, United States) • Bachelor's Degree • Internship • Intern
Python
C++
C++
PyTorch C++
AI/ML
JAX
Multimodal AI
PyTorch
World Models
Machine Learning
Reinforcement Learning
Computer Vision
Physical AI
Robotics
Reinforcement Learning
+4
Apply
Top 25% payVerified live · 1 hour ago 4 months ago Fading · 25%

Senior Machine Learning Engineer

$208k – $264k per year • Equity • Senior • 4+ years exp • In office (San Francisco, United States) • Senior
Python
C++
C++
PyTorch C++
AI/ML
CUDA Toolkit
JAX
Multimodal AI
Knowledge Distillation
Diffusion Models
TensorRT
Transformers
PyTorch
Tokenization
CUDA
Post-training
Vision-Language-Action
World Models
Model Distillation
Machine Learning
Physical AI
Robotics
Sensor Fusion
+12
Apply
Top 25% payVerified live · 1 hour ago 4 months ago Fading · 26%

Staff Machine Learning Engineer

$273k – $345k per year • Equity • Staff+ • 10+ years exp • In office (San Francisco, United States) • Staff
Python
C++
C++
PyTorch C++
AI/ML
CUDA Toolkit
JAX
Multimodal AI
Knowledge Distillation
Diffusion Models
TensorRT
Transformers
PyTorch
Tokenization
CUDA
Post-training
Vision-Language-Action
World Models
Model Distillation
Machine Learning
Physical AI
Robotics
Sensor Fusion
+12
Apply

OpenMind

openmind.com
openmind.com • HQ: San Francisco, United States • Robotic Software & Control Systems • AI Agents • Startup • Hiring 7
HQ: San Francisco, United States • Robotic Software & Control Systems • AI Agents • Startup • Hiring 7
OpenMind is the AI brain and infrastructure layer for teams of robots in human-facing environments such as factories, retail stores, hospitals and the field. It builds the models and operating system that let robots understand people, context and norms and runs them on real robots with customers, and it hires machine learning, robotics, reliability and forward deployed engineers. The company is headquartered in San Francisco, California, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago 12 days ago Hiring now · 70%

Machine Learning Engineer

$130k – $180k per year • Junior • 1+ year exp • In office (San Francisco, United States) • Master's Degree • Full-Time • Junior
Python
C++
C++
PyTorch C++
AI/ML
JAX
ONNX
TensorRT
PyTorch
Machine Learning
DevOps
Docker
Robotics
ROS2
Imitation Learning
+3
Apply

Waymo

waymo.com
waymo.com • HQ: Mountain View, United States • Ride Hailing • Smart Mobility • Autonomous Driving • 5000+ employees • Est. 2009 • Trust B • Hiring 351, 17 new
HQ: Mountain View, United States • Ride Hailing • Smart Mobility • Autonomous Driving • 5000+ employees • Est. 2009 • Trust B • Hiring 351, 17 new
Waymo is an American autonomous driving company that began as the Google self-driving car project in 2009 and became a standalone Alphabet subsidiary in 2016. It operates the largest fully driverless commercial ride-hailing service in the world through the Waymo One app, carrying paying passengers with no human in the vehicle across Phoenix, San Francisco, Los Angeles, Austin and an expanding list of cities. Headquartered in Mountain View, California, the company builds its own sensor suite and the Waymo Driver software stack, integrates them into vehicles from partners such as Jaguar, Zeekr and Hyundai, and also develops autonomous trucking technology.
Dmitri Dolgov · Co-Chief Executive Officer & Chief Technology Officer · decision-maker
Tekedra Mawakana · Co-Chief Executive Officer · decision-maker
Sebastian Thrun · Founder of the Google Self-Driving Car Project · founder
Anthony Levandowski · Founding Engineer
No careers mailbox known
Top 25% payVerified live · 3 hours ago Visa sponsor 12 days ago Hiring now · 90%

Senior Software Engineer, Vehicle Interaction

$213k – $263k per year • Equity • Senior • 5+ years exp • Hybrid (Mountain View, United States) • Bachelor's Degree • Full-Time • Senior
C++
C++
PyTorch C++
AI/ML
JAX
PyTorch
Machine Learning
Robotics
Motion Planning
Imitation Learning
Apply
Top 25% payVerified live · 3 hours ago Visa sponsor 22 days ago Hiring now · 89%

2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning

$140k per year • Intern • In office (Mountain View, United States) • Relocation • Master's Degree • Internship • Intern
Python
AI/ML
LoRA
Fine-tuning
Quantization
JAX
Multimodal AI
Knowledge Distillation
Computer Vision
QLoRA
Transformers
TensorFlow
PyTorch
Vision Transformer
Flax
TPU
Model Distillation
Machine Learning
PEFT
Robotics
Sensor Fusion
+11
Apply
Top 25% payVerified live · 3 hours ago Visa sponsor 23 days ago Hiring now · 89%

2027 Summer Intern, PhD, Software Engineer, Simulation

$170k per year • Intern • In office (Mountain View, United States) • PhD • Internship • Intern
Python
AI/ML
JAX
TensorFlow
TensorBoard
Machine Learning
Apply

Anduril Industries

anduril.com
anduril.com • HQ: Costa Mesa, United States • Defense Manufacturing • C4ISR • Military Drones • Defense AI • 5000+ employees • Est. 2017 • Trust B • Hiring 2067, 115 new
HQ: Costa Mesa, United States • Defense Manufacturing • C4ISR • Military Drones • Defense AI • 5000+ employees • Est. 2017 • Trust B • Hiring 2067, 115 new
Anduril Industries is an American defence technology company founded in 2017 that builds autonomous systems and command software for military and border security customers. Rather than bidding on cost-plus programmes, it develops products with its own capital and sells them as finished systems, from surveillance towers and counter-drone interceptors to underwater vehicles and collaborative combat aircraft. Its Lattice software platform fuses sensor data into a single operating picture and controls the hardware family, and the company is headquartered in Costa Mesa, California with a large manufacturing site in Ohio.
Matt Grimm · Co-Founder & Chief Operating Officer · decision-maker
Trae Stephens · Co-Founder & Executive Chairman · decision-maker
Brian Schimpf · Co-Founder & Chief Executive Officer · decision-maker
Palmer Luckey · Founder · decision-maker
Careers mailbox is known
Top 25% payVerified live · 2 hours ago 12 days ago Likely open · 63%

Senior Computer Vision Engineer

$220k – $292k per year • Equity • Senior • 6+ years exp • Irvine • Remote (United States) • Bachelor's Degree • Secret • Senior
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
JAX
Computer Vision
ONNX
TensorRT
TensorFlow
PyTorch
Edge AI
Machine Learning
DevOps
CI/CD
Linux
Robotics
Sensor Fusion
SLAM
+8
Apply
Verified live · 2 hours ago 2 months ago Long shot · 23%

Staff Gen AI Research Scientist

≈ $190k – $388k per year (Estimated) • Equity • Staff+ • 2+ years exp • In office (Costa Mesa, Washington, Seattle, United States) • Bachelor's Degree • Top Secret • Staff
Python
AI/ML
LangChain
DeepSpeed
LlamaIndex
Fine-tuning
RLHF
Quantization
JAX
Multimodal AI
Knowledge Distillation
Computer Vision
AI Agents
NLP
Axolotl
PyTorch
LLM
Hugging Face
DPO
SFT
Post-training
Megatron-LM
Model Distillation
Machine Learning
Reinforcement Learning
Diffusion Models
VLM
TPU
RLAIF
Robotics
SLAM
Sensor Fusion
Motion Planning
Reinforcement Learning
+24
Apply
rbc.com • HQ: Toronto, Canada • Investment Banking & M&A Advisory • Wealth Management & Financial Advisors • Commercial & Retail Banks • 1001-5000 employees • Est. 1864 • NYSE: RY • Trust A • Hiring 305, 64 new
HQ: Toronto, Canada • Investment Banking & M&A Advisory • Wealth Management & Financial Advisors • Commercial & Retail Banks • 1001-5000 employees • Est. 1864 • NYSE: RY • Trust A • Hiring 305, 64 new
Headquartered in Toronto, Ontario - with its registered head office in Montreal, Quebec - Royal Bank of Canada (RBC) is a major global financial services enterprise and Canada's largest bank by market capitalization. The institution operates a diversified universal banking platform spanning personal and commercial banking, global wealth management, capital markets, insurance, and investor services. Serving over 19 million consumer, commercial, and institutional clients across nearly 30 countries, it functions as a global systemically important bank (G-SIB) and a primary provider of international financial infrastructure.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago Visa sponsor 12 days ago Likely open · 49%

2027 Winter – GRM, AI Developer Intern - Innovation & AI Center of Excellence (4 Months)

Intern • Hybrid (Toronto, Canada) • Bachelor's Degree • Internship • Intern
Python
Databases
Weaviate
pgvector
PostgreSQL
AI/ML
LangGraph
LangChain
LlamaIndex
Prompt Engineering
AI Agents
Haystack
Transformers
LLM
RAG
Hugging Face
Machine Learning
Model Context Protocol
Fine-tuning
Scikit-learn
JAX
TensorFlow
PyTorch
Chips/EDA
PoC Library
+14
Apply

Incluso

incluso.se
incluso.se • HQ: Gothenburg, Sweden • Recruiting & Staffing • Hiring 4, 2 new
HQ: Gothenburg, Sweden • Recruiting & Staffing • Hiring 4, 2 new
Incluso is a Swedish recruitment and consultancy staffing firm focused on international talent. It places engineers, designers, analysts, logistics coordinators and HR specialists with clients such as a global automotive company, mostly in Gothenburg and Stockholm. The firm is based in Gothenburg, Sweden.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 12 days ago Fading · 27%

Wireless AI Algorithm Engineer

In office (Stockholm, Sweden) • PhD • Full-Time • English
Python
AI/ML
Reinforcement Learning
JAX
Transfer Learning
TensorFlow
PyTorch
Edge AI
Machine Learning
Apply
Machines can read this selection · checked 14:30 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.