Updated Aug 31, 2026

JAX Jobs - Remote & On-site

JAX jobs at vetted startups and product companies focused on high-performance ML and research, updated daily. These roles often pay competitively and support remote or hybrid work arrangements. Browse and apply today.

Open positions
502
Companies
235
Salary range
$165K – $450K
Average salary
$286K
AI/ML
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Executive
Networking
Product
Design
Analytics
Support
Enterprise Apps
Quantum

Accenture

accenture.com
Accenture is a leading global professional services company specializing in strategy, consulting, technology, and operations services. Headquartered in Dublin, Ireland, the firm helps organizations optimize their business performance, navigate digital transformation, and implement advanced technologies like AI and cloud computing.
accenture.com • HQ: Dublin, Ireland • HQ: Chicago, United States • Artificial Intelligence • IT Consulting • Information Technology • 5000+ employees • Est. 1989
HQ: Dublin, Ireland • HQ: Chicago, United States • Artificial Intelligence • IT Consulting • Information Technology • 5000+ employees • Est. 1989
Verified live · 2 hours ago 5 hours ago

^Technical Program Manager - 6457364

$215k – $393k per year (Estimated) • Staff+ • 10+ years expRemote/Hybrid (San Francisco, Mountain View, United States) • Visa sponsorship • High School Diploma • Full-Time • Staff
JAX
TensorFlow
TPU
Apply
Report

Citizens Financial Group

citizensbank.com
Citizens Financial Group is one of the oldest and largest financial institutions in the United States, offering a comprehensive suite of commercial and retail banking services. Headquartered in Providence, Rhode Island, the enterprise operates an extensive network of physical branches, ATMs, and digital platforms across multiple regions. Through its tailored personal, private, and corporate solutions, the bank assists individuals, small businesses, and institutions in managing capital and achieving their financial goals.
citizensbank.com • HQ: Providence, United States • Financial Services • Banking • 501-1000 employees
HQ: Providence, United States • Financial Services • Banking • 501-1000 employees
Verified live · 1 hour ago 1 day ago

Principal Software Engineer

$108k – $242k per year (Estimated) • Lead • 3+ years exp • Remote (United States) • High School Diploma • Principal
Java
SQL
JavaScript
Java
Hibernate
Maven
Spring Boot
Spring MVC
Databases
Apache Kafka
AI/ML
JAX
Frontend
Bootstrap
Mobile
JUnit
DevOps
Bitbucket
CI/CD
Datadog
Docker
Git
Kubernetes
Logstash
Rest API
ZooKeeper
QA
Swagger
Apply
Report

Uber

uber.com
Uber Technologies, Inc. is a global transportation and mobility platform headquartered in San Francisco, California. Founded in 2009 by Garrett Camp and Travis Kalanick, the company revolutionized urban transit by pioneering ride-hailing via a smartphone app.
uber.com • HQ: San Francisco, United States • Transportation & Logistics • Delivery • Ride Hailing • 1001-5000 employees • Est. 2009
HQ: San Francisco, United States • Transportation & Logistics • Delivery • Ride Hailing • 1001-5000 employees • Est. 2009
Verified live · 1 hour ago 1 day ago

Staff iOS Engineer

$47k – $109k per year (Estimated) • Staff+ • 10+ years expIn office (Bengaluru, India) • Bachelor's Degree • Staff
SQL
AI/ML
AI Agents
JAX
Knowledge Distillation
NLP
PyTorch
Scikit-learn
TensorFlow
Apply
Top 25% payVerified live · 1 hour ago 29 days ago

Senior Staff Machine Learning Engineer - Core Services Engineering

$267k – $297k per year • Equity • Staff+ • 10+ years expIn office (Sunnyvale, United States) • Master's Degree • Staff
Java
Python
SQL
Databases
Presto
AI/ML
JAX
PyTorch
Spark
TensorFlow
Anomaly Detection
Apply
Top 25% payVerified live · 1 hour ago 1 month ago

Senior ML Engineer, Computer Vision - Applied AI

$202k – $224k per year • Equity • Senior • 5+ years expIn office (San Francisco, United States) • Senior
Python
AI/ML
Computer Vision
JAX
Multimodal AI
PyTorch
Quantization
TensorFlow
ONNX
OCR
Apply
Report

Leidos

leidos.com
Leidos is a defense, intelligence, civil, and health information technology enterprise. Headquartered in Reston, Virginia, the Fortune 500 company (NYSE: LDOS) operates as one of the primary IT, scientific research, and systems integration contractors for the U.S. federal government, allied defense agencies, and commercial infrastructure entities.
leidos.com • HQ: Reston, United States • Government • Cybersecurity • Military • 5000+ employees • Est. 2013
HQ: Reston, United States • Government • Cybersecurity • Military • 5000+ employees • Est. 2013
Top 25% payVerified live · 1 day ago 18 days ago

Senior Software Engineer

$154k – $278k per year • Senior • In office (Annapolis Junction, United States) • Master's Degree • Full-Time • Senior
C++
Groovy
JavaScript
Python
SQL
TypeScript
Java
C
Kotlin
C++
CMake
Java
Jackson
Spring Framework
Apache Tomcat
Maven
C
Embedded C
Kotlin
Mockito
Databases
PostgreSQL
AI/ML
JAX
Frontend
Angular
Bootstrap
JQuery
Mobile
JUnit
DevOps
Configuration Management
CI/CD
Docker
Jenkins
Kubernetes
OpenShift
Podman
QA
Selenium
Apply
Top 25% payVerified live · 1 day ago 21 day ago

Senior Software Developer

$108k – $195k per year • Senior • 8+ years expIn office (Chantilly, United States) • Full-Time • Senior
Bash
JavaScript
Python
SQL
Databases
ElasticSearch
Oracle
AI/ML
JAX
Frontend
Bootstrap
Vue.js
Vuetify
Mobile
Material Design
DevOps
AWS
Git
Jenkins
Kibana
Logstash
Nagios
Puppet
GitHub
Apply
Verified live · 1 day ago 1 month ago

Software Engineer TS/SCI Polygraph

$58k – $105k per year • Junior • 2+ years expIn office (Annapolis, United States) • Full-Time • Junior
C++
Groovy
JavaScript
Python
SQL
TypeScript
Java
C
Kotlin
C++
CMake
Java
Jackson
Spring Framework
Apache Tomcat
Maven
C
Embedded C
Kotlin
Mockito
Databases
PostgreSQL
AI/ML
JAX
Frontend
Angular
Bootstrap
JQuery
Mobile
JUnit
DevOps
Configuration Management
CI/CD
Docker
Jenkins
Kubernetes
OpenShift
Podman
QA
Selenium
Apply
Report
The PNC Financial Services Group is one of the largest diversified banking and financial institutions in the United States. Headquartered in Pittsburgh, Pennsylvania, the corporation provides a comprehensive range of services, including retail banking, wealth management, and corporate lending. With a vast network of branches and ATMs, it serves millions of individual consumers, small businesses, and large corporate clients across the country.
pnc.com • HQ: Pittsburgh, United States • Health Care • Financial Services • Banking • 1001-5000 employees
HQ: Pittsburgh, United States • Health Care • Financial Services • Banking • 1001-5000 employees
Verified live · 1 day ago 8 days ago

Software Engineer Principal

$118k – $240k per year (Estimated) • Lead • 5+ years expIn office (Pittsburgh, United States) • Bachelor's Degree • Full-Time • Principal
Java
SQL
Java
Hibernate
Databases
Oracle
AI/ML
JAX
DevOps
Bitbucket
CI/CD
Jenkins
QA
SoapUI
Apply
Report

Brave

brave.com
Brave Software is an internet browser that is focused in on being privacy-oriented and is combined with its blockchain-based digital advertising platform. The company is based out of San Francisco, California.
brave.com • London • Search Engines • Artificial Intelligence • 501-1000 employees
London • Search Engines • Artificial Intelligence • 501-1000 employees
Verified live · 3 hours ago 2 days ago

AI Security Researcher/Research Engineer

$103k – $216k per year (Estimated)In office (London, United Kingdom) • PhD
Python
AI/ML
Fine-tuning
JAX
LLM
PyTorch
LLM Guardrails
Cybersecurity
Least Privilege
Apply
Report

Jump Trading

jumptrading.com
Jump Trading is a proprietary trading firm focused on quantitative and high-speed strategies. It invests heavily in networking, hardware acceleration and research computing. The firm also runs ventures in cryptocurrency infrastructure and machine learning research.
jumptrading.com • HQ: Chicago, United States • Est. 1999
HQ: Chicago, United States • Est. 1999
Verified live · 4 hours ago 8 days ago

Campus AI/ML Researcher (Intern)

Intern • In office (Hong Kong) • Internship • Intern
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
PyTorch
TensorFlow
Apply
Verified live · 4 hours ago 13 days ago

Research Scientist/Research Engineer, Reinforcement Learning

$116k – $228k per year (Estimated) • Senior • 5+ years expIn office (Chicago, United States) • Master's Degree • Senior • English
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
PyTorch
Reinforcement Learning
TensorFlow
Robotics
Reinforcement Learning
Apply
Top 25% payVerified live · 4 hours ago 1 month ago

Campus AI Research Engineer – Deep Learning (Full-Time)

$300k per year • In office (Chicago, United States) • Visa sponsorship • Full-Time
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
CUDA
CUDA Toolkit
JAX
PyTorch
TensorFlow
ROCm
DevOps
HPC
Apply
Report

Hestus

hestus.co
Hestus is AI-powered CAD software that speeds up hardware development.
hestus.co • San Mateo • CAD & PLM • Artificial Intelligence • 11-50 employees
San Mateo • CAD & PLM • Artificial Intelligence • 11-50 employees
Top 25% payVerified live · 1 hour ago 4 days ago

Machine Learning Engineer

$180k – $240k per year • Equity • Senior • 6+ years expIn office (San Mateo, United States) • Master's Degree • Full-Time • Senior
Python
SQL
Python
Django
Flask
AI/ML
Embeddings
JAX
LLM
PyTorch
TensorFlow
Amazon SageMaker
DevOps
AWS
Azure
GCP
Apply
Report

Target

target.com
Target Corporation is an American retail corporation headquartered in Minneapolis, Minnesota, and was founded in 1902 as the Dayton Dry Goods Company. The company operates a vast network of large-scale discount department stores and hypermarkets that sell a wide range of products including apparel, electronics, home decor, groceries, and health and beauty items. As one of the largest retailers in the United States, it manages nearly 2,000 store locations across the country and maintains a significant e-commerce presence through its website and mobile application.
target.com • HQ: Minneapolis, United States • Food & Beverages • Home Decor • Apparel & Fashion • Consumer Goods • Online Groceries • Commerce • Retail & Shop Systems • 5000+ employees • Est. 1902
HQ: Minneapolis, United States • Food & Beverages • Home Decor • Apparel & Fashion • Consumer Goods • Online Groceries • Commerce • Retail & Shop Systems • 5000+ employees • Est. 1902
Top 25% payVerified live · 21 min ago 20 days ago

Sr Applied Data Scientist - Personalization (applied ML, PyTorch, ML Ops)

$98k – $176k per year • Senior • 3+ years expIn office (Brooklyn Park, United States) • Master's Degree • Full-Time • Senior
Python
SQL
AI/ML
JAX
Spark
Apply
Report

Torus

usetorus.com
AI agents that think in P&IDs, HMBs, and spec sheets. Torus understands every parameter across your projects and adopts how your firm works.
usetorus.com • San Francisco • AI Agents • Network Security
San Francisco • AI Agents • Network Security
Top 25% payVerified live · 2 hours ago 4 days ago

AI Engineer (interested in physical infrastructure)

$150k – $250k per year • Equity 0.2–0.5% • Junior • 1+ year expIn office (San Francisco, United States) • Full-Time • Junior
C++
Python
Rust
TypeScript
C++
PyTorch C++
AI/ML
JAX
LLM
PyTorch
RAG
AI Agents
Function Calling
Apply
Report

Ciandt

ciandt.com
Ciandt is a leading provider of digital transformation services. We help organizations transform their businesses by leveraging the latest technologies. Our solutions provide comprehensive digital strategy, design, development, and implementation capabilities.
ciandt.com • HQ: Campinas, Brazil • IT Consulting • Software • Information Technology • 5000+ employees • Est. 1995
HQ: Campinas, Brazil • IT Consulting • Software • Information Technology • 5000+ employees • Est. 1995
$39k – $111k per year (Estimated) • Senior • Remote (Brazil) • Middle • Portuguese: C1
Java
SQL
Kotlin
Java
Gradle
Hibernate
Maven
Spring Boot
Spring MVC
Spring Security
Kotlin
Mockito
Databases
Apache Kafka
AI/ML
JAX
Mobile
JUnit
DevOps
Azure
Bamboo
Bitbucket
CI/CD
Docker
Dynatrace
Git
Kubernetes
Cybersecurity
Fortify
SonarQube
Management
Confluence
Jira
Apply
$40k – $115k per year (Estimated) • Senior • Brazil • Remote (Brazil) • Full-Time • Senior • Portuguese: C1
Java
Java
Spring Boot
AI/ML
JAX
DevOps
AWS
Azure
Docker
Apply
Verified live · 6 min ago 1 month ago

[Job-30597] Systems Architect, Brazil

$67k – $140k per year (Estimated) • Staff+ • Remote (Brazil) • Architect • Portuguese: C1
Java
Node JS
Kotlin
JavaScript
Java
Gradle
Hibernate
Maven
OpenJDK
Quarkus
Spring Boot
Kotlin
Mockito
Databases
ActiveMQ
PostgreSQL
AI/ML
JAX
Mobile
JUnit
DevOps
AWS
AWS Lambda
Azure
Azure DevOps
CI/CD
Docker
OpenShift
Amazon S3
Cybersecurity
SonarQube
Veracode
Apply
Report

Nanimai

nanimai.tech
Nanimai is an AI-driven HRTech platform designed to automate and streamline mass and volume talent acquisition workflows. The SaaS system features 24/7 conversational AI chatbots that conduct candidate pre-qualification, handle automated screening interviews, answer applicant FAQs, and schedule follow-ups. By aggregating resumes from major job boards, performing outbound sourcing across public web sources, and integrating directly into enterprise Applicant Tracking Systems (ATS), it centralizes recruitment pipelines into a single interface.
nanimai.tech • Kazan
Kazan
Remote (EAEU) • Russian: C1
Python
AI/ML
Chain-of-Thought
GigaChat
JAX
LLM
Mistral
NLP
Prompt Engineering
PyTorch
RAG
Semantic Search
TensorFlow
YandexGPT
Anthropic
OpenAI
Semantic Search
Function Calling
Apply
Report

Sereact

sereact.ai
Sereact is a Stuttgart company founded in 2021 that builds artificial intelligence software for robotic picking. Its models let robots grasp objects they have never seen before and follow instructions given in natural language. The company works with logistics and manufacturing customers automating handling tasks.
sereact.ai • HQ: Stuttgart, Germany • Transportation & Logistics • Artificial Intelligence • Warehousing • Est. 2021
HQ: Stuttgart, Germany • Transportation & Logistics • Artificial Intelligence • Warehousing • Est. 2021
Verified live · 8 hours ago 4 days ago

Senior Robotics Engineer - R&D

$126k – $238k per year (Estimated) • Senior • In office (Boston, United States) • Master's Degree • Full-Time • Senior
Computer Vision
Diffusion Models
Fine-tuning
JAX
Multimodal AI
PyTorch
Synthetic Data
Transformers
Apply
Verified live · 8 hours ago 11 days ago

Senior Research Scientist

$146k – $283k per year (Estimated) • Senior • In office (Boston, United States) • Master's Degree • Full-Time • Senior
Computer Vision
Diffusion Models
Fine-tuning
JAX
Multimodal AI
PyTorch
Synthetic Data
Transformers
Apply
Verified live · 8 hours ago 1 month ago

Senior Research Scientist (m/f/d)

$107k – $270k per year (Estimated) • Senior • In office (Zurich, Switzerland) • Relocation • Master's Degree • Full-Time • Senior
Computer Vision
Diffusion Models
Fine-tuning
JAX
Multimodal AI
PyTorch
Synthetic Data
Transformers
Apply
Report

Platonus

platonus.kz
Лидер цифровой трансформации образования в Казахстане: Platonus Group - надежный технологический партнер в связке образования, молодежи и будущего государства.
platonus.kz • Astana
Astana
Verified live · 8 hours ago 4 days ago

Тестировщик QA

In office (Astana, Kazakhstan) • Russian: C1 • Kazakh (Optional)
Java
Python
SQL
JavaScript
Kotlin
TypeScript
Java
Apache Tomcat
Maven
Kotlin
Mockito
Databases
MySQL
AI/ML
JAX
Frontend
Angular
Angular Material
DevOps
Git
Jenkins
Rest API
Docker
Analytics
Power BI
Management
Confluence
Jira
QA
Insomnia
Playwright
Postman
Pytest
Selenium
Swagger
TestNG
JMeter
k6
Apply
Report

Axiommath

axiommath.ai
The Starting Point for Reasoning.. Test verify_proof, extract theorems, and experiment with proof transformations in seconds.
Open 149 days 4 months ago

Member of Technical Staff

$169k – $342k per year (Estimated) • Staff+ • In office • Full-Time • Staff
Python
AI/ML
JAX
PyTorch
Reinforcement Learning
TensorFlow
Apply
Report

Tzafon

tzafon.ai
Tzafon is advancing machine intelligence. Our team develops models and systems for autonomous agents interacting with the real world.
tzafon.ai • San Francisco • AI Agents
San Francisco • AI Agents
Open 325 daysTop 25% pay 10 months ago

Member of Technical Staff - Foundations

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

Thinking Machines Lab

thinkingmachines.ai
Thinking Machines Lab is an artificial intelligence research and product company based in San Francisco and founded in 2025. The company develops multimodal AI systems and open-weights models, such as Inkling, alongside developer tools like Tinker for model fine-tuning. It operates as a public benefit corporation focused on human-AI collaboration and open science, supported by significant venture capital investment.
thinkingmachines.ai • HQ: San Francisco, United States • MLOps • Machine Learning • Multimodal AI • LLM & Generative AI • Artificial Intelligence • 201-500 employees • Est. 2025
HQ: San Francisco, United States • MLOps • Machine Learning • Multimodal AI • LLM & Generative AI • Artificial Intelligence • 201-500 employees • Est. 2025
Verified live · 1 hour ago 5 days ago

Research, Post-Training Evals

$232k – $508k per year (Estimated)Remote/Hybrid (San Francisco, United States) • PhD • Full-Time
Python
AI/ML
Post-training
AI Agents
JAX
LLM
PyTorch
Reinforcement Learning
TensorFlow
Apply
Top 25% payVerified live · 1 hour ago 27 days ago

Research, Pre-Training Science

$350k – $475k per year • Remote/Hybrid (San Francisco, United States) • Relocation • Visa sponsorship • PhD • Full-Time
Python
AI/ML
JAX
PyTorch
TensorFlow
Pre-training
Apply
Top 25% payVerified live · 1 hour ago 27 days ago

Research, Pre-Training Data

$350k – $475k per year • Remote/Hybrid (San Francisco, United States) • Relocation • Visa sponsorship • PhD • Full-Time
Python
AI/ML
JAX
Multimodal AI
PyTorch
TensorFlow
Human-in-the-Loop
Pre-training
Apply
Report

xAI

x.ai
xAI is an American artificial intelligence company founded by Elon Musk in 2023 with the stated goal of building models that help humans understand the universe. It develops the Grok family of large language models, distributes them through a consumer assistant, a developer API and deep integration with the X social platform, and adds image and video generation through Grok Imagine. The company runs its own Colossus supercomputer clusters in Memphis, Tennessee, is headquartered in Palo Alto, California, and merged with X Corp in 2025 to combine model development with a large consumer distribution channel.
x.ai • HQ: Palo Alto, United States • Multimodal AI • AI Agents • Machine Learning • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2023
HQ: Palo Alto, United States • Multimodal AI • AI Agents • Machine Learning • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2023
Top 25% payVerified live · 2 hours ago 4 months ago

Member of Technical Staff - Multimodal Understanding

$440k per year • Staff+ • In office (Palo Alto, United States) • Staff
Python
C++
Rust
C++
PyTorch C++
AI/ML
Fine-tuning
JAX
Multimodal AI
PyTorch
Tokenization
Post-training
Pre-training
SFT
TPU
AI Agents
Function Calling
Ray
Spark
DevOps
Kubernetes
Apply
Top 25% payVerified live · 2 hours ago 5 months ago

Member of Technical Staff - Imagine Model

$440k per year • Staff+ • In office (Palo Alto, United States) • Staff
C++
Python
Rust
C++
PyTorch C++
AI/ML
Knowledge Distillation
Multimodal AI
Edge AI
Post-training
AI Agents
JAX
PyTorch
Ray
Spark
Synthetic Data
SFT
Apply
Top 25% payVerified live · 2 hours ago 5 months ago

Software Engineer - Voice Model

$450k per year • In office (Palo Alto, United States)
Python
AI/ML
Fine-tuning
JAX
PyTorch
Ray
Reinforcement Learning
Spark
Synthetic Data
Post-training
Pre-training
SFT
DevOps
Kubernetes
Analytics
A/B Testing
Apply
Report

Citi

citi.com
Citi is a leading global financial services company headquartered in New York City, offering a broad range of banking, investment, and wealth management services to consumers, corporations, and institutions. As one of the world's largest banking institutions, it operates across more than 160 countries and jurisdictions, serving as a critical facilitator of global commerce. Through its primary consumer division, Citibank, alongside its institutional businesses, the company provides everyday banking, credit cards, capital markets solutions, and cross-border payment infrastructure worldwide.
citi.com • HQ: New York, United States • Credit Cards • Financial Services • Banking • 1001-5000 employees • Est. 1900
HQ: New York, United States • Credit Cards • Financial Services • Banking • 1001-5000 employees • Est. 1900
Verified live · 2 hours ago 5 days ago

JAVA Microservices Engineering Lead

$121k – $171k per year • Lead • 6+ years expIn office (Mississauga, Canada) • Bachelor's Degree • Full-Time • Staff
Java
TypeScript
JavaScript
COBOL
Java
Spring Boot
Spring Cloud
Spring Security
Hazelcast
COBOL
IBM MQ
Databases
Apache Kafka
Memcached
RabbitMQ
Redis
AI/ML
JAX
Frontend
Angular
DevOps
AWS
Azure
CI/CD
Configuration Management
GCP
Git
Gitflow
Jenkins
Kubernetes
OpenShift
Tekton
AppDynamics
Dynatrace
API Gateway
Apply
Top 25% payVerified live · 2 hours ago 6 days ago

Java Apps Developer Lead Analyst - Vice President

$142k – $213k per year • Executive • 6+ years expRemote/Hybrid (Jersey City, New York, United States) • Bachelor's Degree • Full-Time • Staff
Java
SQL
Kotlin
Java
Hibernate
Jackson
Spring Framework
Kotlin
Mockito
Databases
Apache Kafka
AI/ML
JAX
Mobile
JUnit
DevOps
Docker
Git
Kubernetes
Apply
Verified live · 2 hours ago 14 days ago

Python Engineer - Assistant Vice President

$107k – $161k per year • Executive • 5+ years expRemote/Hybrid (Irving, United States) • Bachelor's Degree • Full-Time • Architect
Python
AI/ML
Google ADK
JAX
PyTorch
TensorFlow
DevOps
AWS
Azure
Docker
GCP
Kubernetes
Apply
Report

Symbotic

symbotic.com
Symbotic is a global provider of integrated supply network automation solutions for warehouses and distribution centers. Our Symbotic Employees build hardware and software solutions that will change the way the world distributes product from manu...
symbotic.com • HQ: Wilmington, United States • Warehousing • Robotics • Logistics & Warehouse Robotics • 1001-5000 employees
HQ: Wilmington, United States • Warehousing • Robotics • Logistics & Warehouse Robotics • 1001-5000 employees
Top 25% payVerified live · 2 days ago 5 days ago

Senior Robotics Engineer - Robot Perception

$149k – $205k per year • Senior • 8+ years expIn office (Wilmington, United States) • Bachelor's Degree • Full-Time • Senior
C#
C++
Python
C++
PyTorch C++
AI/ML
Computer Vision
Data Augmentation
Gaussian Splatting
Synthetic Data
JAX
PyTorch
Robotics
Localization
Perception
Sim-to-Real
Sensor Fusion
SLAM
Apply
Report

Salary range

Seniority Jobs 25% Median 75%
Junior 14 $196K $250K $300K
Middle 19 $181K $252K $300K
Senior 45 $204K $242K $288K
Staff 35 $230K $300K $325K
Lead 18 $220K $270K $286K
All levels 189 $220K $280K $350K

Based only on the listings that state pay. Gross annual amounts, converted to USD so roles in different currencies stay comparable.

Fully remote 15%
73
Hybrid 22%
108
On site 64%
321
Visa sponsorship 12%
60
Relocation package 11%
56
Equity 15%
74

The market right now

Alion currently lists 502 open JAX jobs from 235 companies. 43 of them were posted or refreshed in the last seven days. 179 of the employers have posted something in the last three months, which is the pool worth watching if you are starting a search now. Every listing links straight to the employer, so you apply on their own board rather than through an intermediary.

What these roles pay

189 of these JAX jobs state pay directly. Across them the middle half of the market sits between $220K and $350K a year, with a median of $280K. By level, the median runs junior at $250K, middle at $252K, senior at $242K, staff at $300K, and lead at $270K. Figures are gross annual amounts converted to US dollars, so roles in different currencies stay comparable.

Where the work is

The largest concentrations of these roles are France (27), Poland (15), Switzerland (15), Singapore (14), and United States (13). 15% of the listings are fully remote and a further 22% are hybrid, so a large part of this market is open to you regardless of where you live.

Who is hiring

The employers with the most open JAX jobs right now are Cohere (16), Graphcore (14), NVIDIA (14), Skild (12), Thinking Machines Lab (11), and Lila Sciences (10). Each company page on Alion carries its size, funding stage, tech stack and every other position it has open, so you can judge the employer before you spend an evening on the application.

What employers ask for

Reading across the current listings, the tools that come up most often are JAX (502), PyTorch (455), Python (374), TensorFlow (273), Multimodal AI (148), C++ (144), AI Agents (141), and PyTorch C++ (132). The counts are how many of these openings name each one, which is a better guide to what is actually being hired for than a generic skills list.

Relocation, visas and equity

Of the current JAX jobs, 60 state visa sponsorship, 56 offer a relocation package, and 74 include equity. These are filters on the list above, so you can narrow it to the ones that make a move possible for you.

JAX Jobs - Remote & On-site
Frequently asked questions
How many JAX jobs are open right now?
There are 502 open JAX jobs on Alion from 235 companies. The list was last refreshed on August 31, 2026.
What do JAX jobs pay?
The median is $280K a year, and the middle half of the market falls between $220K and $350K. This is based on the 189 listings that state pay.
Are any of these roles remote?
73 of the 502 listings (15%) are fully remote, and you can filter the list down to them in one click.
Which companies are hiring?
The most active employers right now are Cohere, Graphcore, NVIDIA, Skild, Thinking Machines Lab, and Lila Sciences. Each has its own page on Alion with the rest of its open roles.
Can I get visa sponsorship or relocation?
Yes - of the current listings, 60 state visa sponsorship and 56 offer a relocation package. Both are filters on the jobs page.
How do I apply?
Open any listing and apply on the employer's own board through the link on the page. No account is required to browse or to apply.