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.

AI/ML
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Executive
Networking
Product
Design
Analytics
Support
Enterprise Apps
Quantum

Tenstorrent

tenstorrent.com
Tenstorrent is a Canadian semiconductor company founded in 2016 and led by veteran chip architect Jim Keller. It designs artificial intelligence accelerators and RISC-V processors, and sells intellectual property licences as well as hardware, with an open-source software stack. Investors and licensees include major automotive and technology groups seeking an alternative to closed accelerator ecosystems.
tenstorrent.com • HQ: Toronto, Canada • Embedded Systems • Hardware • Machine Learning • Est. 2016
HQ: Toronto, Canada • Embedded Systems • Hardware • Machine Learning • Est. 2016
Open 1040 daysTop 25% pay 3 years ago

Sr. Software Engineer, AI Compiler

$100k – $500k per year • Senior • In office (Toronto, Canada) • Full-Time • Senior
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
PyTorch
TensorFlow
Edge AI
Human-in-the-Loop
Apply
Open 485 daysTop 25% pay 1 year ago

Sr. Engineer, Software - AI Compiler

$100k – $500k per year • Senior • In office (Austin, United States) • Full-Time • Senior
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
PyTorch
TensorFlow
Edge AI
Human-in-the-Loop
Apply
Open 270 daysVerified live · 28 min ago 9 months ago

Software Engineer-AI Model Bring Up

$41k – $90k per year (Estimated)In office (Tokyo, Japan) • Bachelor's Degree • Full-Time • Japanese
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
CNN
JAX
LLM
PyTorch
TensorFlow
Edge AI
Quantization
Apply
Report

Graphcore

graphcore.ai
Graphcore is a British semiconductor company founded in Bristol in 2016 that designs the Intelligence Processing Unit, a processor architected specifically for machine learning rather than adapted from graphics. Its chips place large amounts of memory directly on the die and expose fine-grained parallelism, an approach aimed at sparse and irregular models that map poorly onto conventional accelerators. The company sells IPU systems and the Poplar software stack to research and enterprise customers, and has operated as a wholly owned subsidiary of SoftBank Group since its acquisition in 2024.
graphcore.ai • HQ: Bristol, United Kingdom • Hardware • Artificial Intelligence • Machine Learning • AI Infrastructure • Semiconductors • 1001-5000 employees • Est. 2016
HQ: Bristol, United Kingdom • Hardware • Artificial Intelligence • Machine Learning • AI Infrastructure • Semiconductors • 1001-5000 employees • Est. 2016
Verified live · 26 min ago 5 days ago

Software Engineer - Triton

$57k – $103k per year (Estimated)In office (Bristol, United Kingdom)
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
PyTorch
TensorFlow
Triton
Apply
Verified live · 26 min ago 4 months ago

Staff Software Engineer (ML QA)

$85k – $185k per year (Estimated) • Staff+ • In office (London, United Kingdom) • Visa sponsorship • Master's Degree • Staff
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
Keras
PyTorch
TensorFlow
DevOps
CI/CD
QEMU
GitHub
Kubernetes
Apply
Verified live · 26 min ago 4 months ago

Staff Software Engineer (ML QA)

$85k – $184k per year (Estimated) • Staff+ • In office (Bristol, United Kingdom) • Visa sponsorship • Master's Degree • Staff
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
Keras
PyTorch
TensorFlow
DevOps
CI/CD
QEMU
GitHub
Kubernetes
Apply
Report

DoorDash

doordash.com
DoorDash is an American technology company headquartered in San Francisco that operates a leading global local commerce and on-demand delivery platform. The company connects consumers with local merchants, offering last-mile logistics for restaurant meals, groceries, convenience goods, and retail items delivered by independent couriers. Beyond its core consumer marketplace, DoorDash provides digital ordering software, merchant advertising tools, and white-label fulfillment services for businesses across multiple countries.
doordash.com • HQ: San Francisco, United States • Commerce • Marketplaces • Transportation & Logistics • Delivery
HQ: San Francisco, United States • Commerce • Marketplaces • Transportation & Logistics • Delivery
Verified live · 2 hours ago 6 months ago

Senior/Staff Deep Reinforcement Learning Engineer - DoorDash Dot

$4k – $14k per year • Equity • Staff+ • In office (San Francisco, United States) • Bachelor's Degree • Staff
C++
AI/ML
Claude
Claude Code
Cursor
JAX
Reinforcement Learning
OpenAI Codex
AI Agents
TensorRT
Robotics
Reinforcement Learning
Sim-to-Real
Apply
Report
General Dynamics Information Technology (GDIT) is a leading technology integrator providing enterprise IT, artificial intelligence, and cloud solutions primarily to government, defense, and intelligence agencies. The company specializes in modernizing complex digital infrastructure, strengthening cybersecurity frameworks, and managing mission-critical systems for public sector clients. Operating as a business unit of defense giant General Dynamics, it plays a key role in driving digital transformation across national security, healthcare, and civil agency operations.
gdit.com • HQ: Falls Church, United States • Military • Government • Information Technology • 1001-5000 employees
HQ: Falls Church, United States • Military • Government • Information Technology • 1001-5000 employees
Top 25% payVerified live · 3 hours ago 6 days ago

Director, Lead Solution Architect AI/ML

$225k – $305k per year • Executive • 15+ years expIn office (Herndon, United States) • Bachelor's Degree • Full-Time • Architect
C++
Java
Python
SQL
C++
PyTorch C++
TensorFlow C++
Databases
Apache Kafka
Databricks
Delta Lake
Google BigQuery
Snowflake
AI/ML
AWS Bedrock
BentoML
DVC
JAX
Kubeflow
LangChain
MLFlow
NumPy
ONNX
PyTorch
Scikit-learn
SciPy
Spark
TensorFlow
Transformers
Vertex AI
Weights & Biases
AI Agents
Amazon SageMaker
Feature Store
Hugging Face
LLM Guardrails
TPU
DevOps
AWS
Azure
Docker
GCP
Helm
Kubernetes
Terraform
IAM
Apply
Report

Oracle

oracle.com
Oracle Corporation is an American multinational computer technology corporation headquartered in Austin, Texas. Founded in 1977 by Larry Ellison, Bob Miner, and Ed Oates, Oracle is one of the world's largest enterprise software and cloud computing infrastructure providers.
oracle.com • HQ: Austin, United States • Software • LLM & Generative AI • Information Technology • Data & Analytics • AI Agents • Artificial Intelligence • Cloud Computing • Enterprise Resource Planning (ERP) • Database Management • 1001-5000 employees • Est. 1977
HQ: Austin, United States • Software • LLM & Generative AI • Information Technology • Data & Analytics • AI Agents • Artificial Intelligence • Cloud Computing • Enterprise Resource Planning (ERP) • Database Management • 1001-5000 employees • Est. 1977
Top 25% payVerified live · 1 hour ago 6 days ago

AI Systems Engineer (OCI/AI Infrastructure)

$97k – $306k per year • Equity • In office (Nashville, United States) • Bachelor's Degree
CUDA
CUDA Toolkit
vLLM
ROCm
JAX
PyTorch
Ray
TensorFlow
NCCL
Apply
Report

Cloudflare

cloudflare.com
Cloudflare is a global web infrastructure and cybersecurity company that provides content delivery network (CDN) services, DDoS mitigation, and edge computing solutions. The platform acts as a protective shield and performance booster between website visitors and origin servers, safeguarding online applications from cyber threats while optimizing speed. Headquartered in San Francisco, it secures and accelerates millions of internet properties and handles a significant portion of all global web traffic.
cloudflare.com • HQ: San Francisco, United States • Information Technology • Cloud Computing • Email Security • Network Security • Cybersecurity • CDN • 1001-5000 employees • Est. 2009
HQ: San Francisco, United States • Information Technology • Cloud Computing • Email Security • Network Security • Cybersecurity • CDN • 1001-5000 employees • Est. 2009
Verified live · 2 hours ago 2 months ago

Senior Machine Learning Engineer

$170k – $329k per year (Estimated) • Senior • Remote/Hybrid • Internship • Senior
Python
AI/ML
Quantization
Embeddings
JAX
llama.cpp
LLM
Multimodal AI
ONNX
PyTorch
SGLang
TensorFlow
TensorRT
TensorRT-LLM
vLLM
Triton
Galileo
RAG
DevOps
Cloudflare
Apply
Report

Lila Sciences

lila.ai
Lila Sciences is an artificial intelligence enterprise developing an autonomous platform designed to accelerate discovery across the life, chemical, and materials sciences. Headquartered in Cambridge, Massachusetts, the company pairs generative AI models with automated robotic laboratories to formulate hypotheses, execute physical experiments, and analyze results. Its integrated system aims to streamline the scientific method, facilitating the rapid creation of novel therapeutics, advanced materials, and industrial compounds for global commercial partners.
lila.ai • HQ: Cambridge, United States • Biotechnology • Software • Operating Systems • 51-200 employees
HQ: Cambridge, United States • Biotechnology • Software • Operating Systems • 51-200 employees
Open 125 daysVerified live · 25 min ago 5 months ago

Senior / Principal ML Scientist, Foundation Models for Life Sciences

$127k – $288k per year (Estimated) • Lead • In office (San Francisco, United States) • Bachelor's Degree • Full-Time • Principal
JAX
Multimodal AI
PyTorch
TensorFlow
Diffusion Models
AI Agents
DevOps
AWS
CI/CD
GCP
Apply
Open 124 daysVerified live · 25 min ago 5 months ago

ML Scientist I / II, Foundation Models for Life Sciences

$118k – $251k per year (Estimated) • Middle • In office (San Francisco, United States) • Master's Degree • Full-Time • Middle
Diffusion Models
JAX
Multimodal AI
PyTorch
TensorFlow
AI Agents
Apply
Open 124 daysVerified live · 25 min ago 5 months ago

Principal, Machine Learning Engineer

$128k – $292k per year (Estimated) • Lead • In office (San Francisco, United States) • PhD • Full-Time • Principal
Diffusion Models
JAX
Multimodal AI
PyTorch
TensorFlow
AI Agents
DevOps
AWS
GCP
Apply
Report

SambaNova Systems

sambanova.ai
SambaNova Systems is an American artificial intelligence semiconductor and cloud platform company headquartered in Palo Alto, California. Founded in 2017 by Stanford professors Kunle Olukotun and Christopher Ré alongside former Oracle executive Rodrigo Liang, the company builds full-stack hardware and software infrastructure purpose-built for enterprise AI, large-scale model training, and high-speed agentic AI inference.
sambanova.ai • HQ: Palo Alto, United States • Artificial Intelligence • Semiconductors • AI Infrastructure
HQ: Palo Alto, United States • Artificial Intelligence • Semiconductors • AI Infrastructure
Open 282 daysVerified live · 4 hours ago 10 months ago

Senior AI Performance Engineer

$151k – $293k per year (Estimated) • Senior • 3+ years expIn office (San Jose, United States) • Bachelor's Degree • Full-Time • Senior
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
DeepSeek
JAX
Llama
PyTorch
Quantization
Qwen
TensorFlow
CUDA
CUDA Toolkit
DeepSpeed
LLM
Multimodal AI
OpenCL
TensorRT
Triton
vLLM
cuDNN
Megatron-LM
Apply
Report

Airwallex

airwallex.com
Airwallex is a global financial technology company that provides cross-border payment, treasury, and banking infrastructure for businesses. Founded in Melbourne in 2015, the platform offers multi-currency business accounts, foreign exchange, payment acceptance, corporate cards, and expense management tools. It helps companies optimize international transactions and scale globally by bypassing traditional, costly banking intermediaries.
airwallex.com • HQ: Singapore • Financial Services • FinTech • Payments • 1001-5000 employees
HQ: Singapore • Financial Services • FinTech • Payments • 1001-5000 employees
Verified live · 1 hour ago 7 days ago

Senior AI Engineer (Risk & Payments)

$96k – $242k per year (Estimated) • Senior • 2+ years expIn office (Sydney, Australia) • Full-Time • Senior
Python
AI/ML
AI Agents
AutoGen
CrewAI
Fine-tuning
JAX
LangChain
LlamaIndex
PyTorch
RAG
Semantic Kernel
Function Calling
Marketing
Salesforce
Apply
Verified live · 1 hour ago 7 days ago

Senior AI Engineer (Risk & Payments)

$107k – $269k per year (Estimated) • Senior • 2+ years expIn office (Singapore) • Full-Time • Senior
Python
AI/ML
AI Agents
AutoGen
CrewAI
Fine-tuning
JAX
LangChain
LlamaIndex
PyTorch
RAG
Semantic Kernel
Function Calling
Marketing
Salesforce
Apply
Top 25% payVerified live · 1 hour ago 3 months ago

Staff AI Engineer

$182k – $310k per year • Staff+ • 5+ years expRemote/Hybrid (San Francisco, United States) • Full-Time • Staff
Python
AI/ML
AI Agents
AutoGen
CrewAI
Fine-tuning
JAX
LangChain
LlamaIndex
PyTorch
RAG
Semantic Kernel
TensorFlow
Function Calling
Marketing
Salesforce
Apply
Report

Salesforce

salesforce.com
Salesforce is a global leader in cloud-based customer relationship management (CRM) software, helping businesses connect with their customers in a whole new way. Its comprehensive Customer 360 platform provides integrated applications for sales, customer service, marketing automation, e-commerce, and analytics.
salesforce.com • HQ: San Francisco, United States • Software • Information Technology • Sales & Marketing • Media & Entertainment • AI Agents • Commerce • Books • Professional Services • Artificial Intelligence • Marketplaces • Marketing Automation • Cloud Computing • Customer Relationship Management (CRM) • 1001-5000 employees
HQ: San Francisco, United States • Software • Information Technology • Sales & Marketing • Media & Entertainment • AI Agents • Commerce • Books • Professional Services • Artificial Intelligence • Marketplaces • Marketing Automation • Cloud Computing • Customer Relationship Management (CRM) • 1001-5000 employees
Verified live · 1 hour ago 8 days ago

AI Scientist — Commerce Labs

$149k – $224k per year • In office (San Francisco, Burlington, Boston, United States) • PhD • Full-Time
Python
AI/ML
AI Agents
JAX
LLM
PyTorch
Reinforcement Learning
TensorFlow
Agentforce
Recommender Systems
DevOps
GitHub
Marketing
Salesforce
Apply
Top 25% payVerified live · 1 hour ago 22 days ago

Staff ML Engineer, Fine Tuning - Slack

$197k – $314k per year • Staff+ • 5+ years expIn office (Washington, Atlanta, San Francisco, United States) • PhD • Full-Time • Staff
Java
PHP
Python
Ruby
Scala
AI/ML
Fine-tuning
JAX
Multimodal AI
NLP
PyTorch
TensorFlow
Agentforce
AI Agents
ONNX
Quantization
RAG
TensorRT
Knowledge Graph
Recommender Systems
Management
Slack
Marketing
Salesforce
Apply
Report

Apptronik

apptronik.com
Apptronik is a humanoid robotics company spun out of a robotics laboratory at the University of Texas at Austin in 2016. Its Apollo robot is designed for logistics and manufacturing work such as case handling and machine tending, with force-controlled actuators built in-house. The company works with Google DeepMind on robot learning and has run pilots with large manufacturers and retailers.
apptronik.com • HQ: Austin, United States • Artificial Intelligence • Design & Creative • Robotics • Est. 2016
HQ: Austin, United States • Artificial Intelligence • Design & Creative • Robotics • Est. 2016
Top 25% payVerified live · 1 day ago 19 days ago

Senior Reinforcement Learning Engineer

$230k – $260k per year • Senior • 5+ years expIn office (Sunnyvale, United States) • Master's Degree • Senior
C++
Python
C++
PyTorch C++
AI/ML
Fine-tuning
JAX
PyTorch
Reinforcement Learning
Robotics
Imitation Learning
Isaac Gym
MuJoCo
Reinforcement Learning
Sim-to-Real
Teleoperation
Apply
Open 185 daysVerified live · 1 day ago 7 months ago

Senior Reinforcement Learning Engineer

$189k – $367k per year (Estimated) • Senior • 5+ years expIn office (Austin, United States) • Master's Degree • Senior
C++
Python
C++
PyTorch C++
AI/ML
Fine-tuning
JAX
PyTorch
Reinforcement Learning
Robotics
Imitation Learning
Isaac Gym
MuJoCo
Reinforcement Learning
Sim-to-Real
Teleoperation
Apply
Verified live · 1 day ago 26 days ago

Reinforcement Learning Engineer

$180k – $338k per year (Estimated) • Middle • 3+ years expIn office (Austin, United States) • PhD • Middle
C++
Python
C++
PyTorch C++
AI/ML
Computer Vision
Fine-tuning
JAX
PyTorch
Reinforcement Learning
Robotics
Imitation Learning
Isaac Gym
MuJoCo
Reinforcement Learning
Sim-to-Real
Teleoperation
Apply
Report

TensorWave

tensorwave.com
TensorWave is an American cloud infrastructure company headquartered in Las Vegas, Nevada, and founded in 2023. The company provides a specialized AI cloud platform powered by AMD Instinct accelerators, offering bare metal instances, managed Kubernetes, and high-speed storage for large-scale model training and inference. It serves enterprise clients and AI research teams globally, positioning itself as a high-performance alternative to NVIDIA-based cloud providers through its focus on open-source ROCm software and cost-efficient scaling.
tensorwave.com • HQ: Las Vegas, United States • Machine Learning • LLM & Generative AI • AI Infrastructure • Hardware • Artificial Intelligence • Information Technology • Cloud Computing • Est. 2023
HQ: Las Vegas, United States • Machine Learning • LLM & Generative AI • AI Infrastructure • Hardware • Artificial Intelligence • Information Technology • Cloud Computing • Est. 2023
Verified live · 1 day ago 7 days ago

Technical Support Engineer

$62k – $144k per year (Estimated) • Equity • In office (Las Vegas, United States) • Full-Time
Bash
Python
AI/ML
CUDA
CUDA Toolkit
JAX
PyTorch
ROCm
DevOps
Kubernetes
PagerDuty
SLURM
SRE
Docker
Grafana
Prometheus
HPC
Apply
Verified live · 1 day ago 3 months ago

Technical Program Manager

$93k – $175k per year (Estimated) • Equity • Middle • 3+ years expIn office (Las Vegas, United States) • Bachelor's Degree • Full-Time • Middle
CUDA
CUDA Toolkit
JAX
PyTorch
ROCm
TPU
DevOps
Kubernetes
Management
Jira
Apply
Report

Instacart AI

instacart.ai
Instacart's enterprise AI platform provides specialized artificial intelligence solutions designed to modernize online and in-store grocery retail. The technology suite includes white-label conversational shopping assistants, computer-vision inventory tracking, and automated catalog enrichment tools trained on extensive historical retail data. By integrating smart hardware and predictive analytics into existing supermarket networks, it enables grocers to streamline operations, enhance inventory visibility, and deliver personalized shopping experiences.
instacart.ai • San Francisco • Chicago • Boston • Wayne • Allentown
San Francisco • Chicago • Boston • Wayne • Allentown
Open 94 daysVerified live · 23 min ago 4 months ago

Senior Machine Learning Engineer II, Ads Response Prediction

$151k – $271k per year (Estimated) • Equity • Senior • Remote (United States) • Master's Degree • Full-Time • Senior
Python
SQL
AI/ML
LLM
Transformers
PEFT
Embeddings
Fine-tuning
JAX
LoRA
Pandas
PyTorch
Ray
Spark
TensorFlow
Transfer Learning
Mixture of Experts
Apply
Report

Harvard University

harvard.edu
Harvard University is a prestigious Ivy League private research institution based in Cambridge, Massachusetts, consistently ranked among the world's most influential universities. Founded in 1636, it is the oldest institution of higher learning in the United States and is renowned for its vast academic programs, rigorous research, and illustrious alumni. Today, the university continues to be a global leader in education and scholarship, maintaining a massive library system and an extensive network of specialized graduate and professional schools.
harvard.edu • HQ: Cambridge, United States • Education • Higher Education
HQ: Cambridge, United States • Education • Higher Education
Verified live · 3 hours ago 8 days ago

Senior Research Software Engineer

$143k – $258k per year (Estimated) • Senior • 7+ years expRemote/Hybrid (Cambridge, United States) • Visa sponsorship • Full-Time • Senior
C++
Python
C++
LLVM
PyTorch C++
TensorFlow C++
AI/ML
CUDA
CUDA Toolkit
DeepSpeed
JAX
ONNX
OpenCL
OpenMP
PyTorch
TensorFlow
TensorRT
Triton
cuDNN
FSDP
ROCm
DevOps
AWS
Git
HPC
Cybersecurity
HIPAA
Apply
Report

Furiosa

furiosa.ai
FuriosaAI designs high-performance, power-efficient AI accelerators (NPUs) used in data centers for computer vision, GenAI, LLMs, and demanding workloads.
furiosa.ai • Seoul • Hwaseong • Santa Clara • San Jose • Data Centers • LLM & Generative AI • Artificial Intelligence
Seoul • Hwaseong • Santa Clara • San Jose • Data Centers • LLM & Generative AI • Artificial Intelligence
Verified live · 2 hours ago 12 days ago

Software Engineer, Machine Learning Engineer (Agentic AI)

In office (Seoul, South Korea)
JAX
LLM
PyTorch
TensorFlow
LangChain
Post-training
SFT
AI Agents
Function Calling
AutoGen
Google ADK
LangGraph
Apply
Report

NVIDIA

nvidia.com
NVIDIA is an American technology company founded in 1993 that invented the graphics processing unit and has become the dominant supplier of accelerated computing platforms for artificial intelligence. Its portfolio spans data centre GPUs and systems built on the Hopper and Blackwell architectures, GeForce consumer graphics, automotive and robotics platforms, high-speed networking acquired with Mellanox, and the CUDA software stack that binds the ecosystem together. Headquartered in Santa Clara, California, the company sells to cloud providers, enterprises, research institutions and gamers worldwide and is one of the most valuable listed businesses on the Nasdaq.
nvidia.com • HQ: Santa Clara, United States • AI Infrastructure • Semiconductors • AI Agents • Big Data • LLM & Generative AI • Machine Learning • Hardware • Data & Analytics • Information Technology • Artificial Intelligence • 5000+ employees • Est. 1993
HQ: Santa Clara, United States • AI Infrastructure • Semiconductors • AI Agents • Big Data • LLM & Generative AI • Machine Learning • Hardware • Data & Analytics • Information Technology • Artificial Intelligence • 5000+ employees • Est. 1993
$184k – $288k per year • Senior • 5+ years expIn office (Santa Clara, United States) • Master's Degree • Full-Time • Senior
Python
AI/ML
CUDA
CUDA Toolkit
Diffusion Models
JAX
LLM
PyTorch
Quantization
TensorRT
Triton
Hugging Face
Apply
Top 25% pay 8 days ago

Solutions Architect, AI Models

$152k – $242k per year • Staff+ • 5+ years expIn office (Santa Clara, United States) • Bachelor's Degree • Full-Time • Architect
Python
AI/ML
JAX
Multimodal AI
PyTorch
Reinforcement Learning
TensorFlow
Transformers
Hugging Face
NVIDIA NeMo
Post-training
Pre-training
DevOps
Kubernetes
SLURM
Apply
$184k – $288k per year • Senior • 8+ years expIn office (Santa Clara, Redmond, Seattle, United States) • Bachelor's Degree • Full-Time • Senior
C++
Go
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
Fine-tuning
JAX
LLM
PyTorch
Ray
TensorFlow
AI Agents
NCCL
DevOps
Kubernetes
Loki
Prometheus
Grafana
Apply
Report

Avride

avride.ai
Avride develops self-driving cars and sidewalk delivery robots, and was created in 2024 when Nebius spun out the former Yandex autonomous vehicle division. Its delivery robots operate on American university campuses and with food delivery platforms, while its robotaxi programme runs with Hyundai and Uber. Engineering is based in Austin, Texas.
avride.ai • HQ: Austin, United States • Transportation & Logistics • Smart Mobility • Autonomous Driving • 201-500 employees • Est. 2024
HQ: Austin, United States • Transportation & Logistics • Smart Mobility • Autonomous Driving • 201-500 employees • Est. 2024
Open 375 daysVerified live · 1 hour ago 1 year ago

Machine Learning Engineer – Motion Planning & Prediction

$138k – $301k per year (Estimated)Austin • Remote (United States) • Relocation
C++
Python
Rust
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
NLP
PyTorch
TensorFlow
AI Agents
Robotics
Motion Planning
SLAM
Apply
Report

Point72

point72.com
Point72 is a global, multi-strategy hedge fund manager founded by Steven A. Cohen that invests across discretionary long/short equity, systematic, and macro strategies. The firm relies heavily on fundamental research, quantitative models, and advanced data analytics to allocate capital across international financial markets. Headquartered in Stamford, Connecticut, it also manages Point72 Ventures, an early-stage venture capital arm focused on investing in emerging technology startups.
point72.com • HQ: Stamford, United States • Venture Capital • Financial Services • Asset Management • 501-1000 employees • Est. 2014
HQ: Stamford, United States • Venture Capital • Financial Services • Asset Management • 501-1000 employees • Est. 2014
Verified live · 21 min ago 1 year ago

Quantitative Researcher - Machine Learning

$127k – $281k per year (Estimated)In office (New York, United States) • Master's Degree
Python
AI/ML
JAX
Reinforcement Learning
TensorFlow
Apply
Report

Sanctuary AI

sanctuary.ai
Sanctuary AI is a humanoid robotics company headquartered in Vancouver, Canada, and founded in 2018 by Geordie Rose, Suzanne Gildert, Olivia Norton, and Ajay Agrawal. The company develops Phoenix, a general purpose humanoid with hydraulically actuated dexterous hands, controlled by a system it calls Carbon that is intended to reason about tasks the way a person would. It runs pilot deployments in retail and industrial settings and has partnered with manufacturers including Magna.
sanctuary.ai • HQ: Vancouver, Canada • Multimodal AI • Robotic Software & Control Systems • Robotics AI • Humanoid Robots • Artificial Intelligence • Robotics • Est. 2018
HQ: Vancouver, Canada • Multimodal AI • Robotic Software & Control Systems • Robotics AI • Humanoid Robots • Artificial Intelligence • Robotics • Est. 2018
Intern • In office (Vancouver, Canada) • Master's Degree • Internship • Intern
Python
AI/ML
Computer Vision
JAX
PyTorch
Reinforcement Learning
Transfer Learning
Robotics
Imitation Learning
Reinforcement Learning
ROS2
Apply
Report

Anthropic

anthropic.com
Anthropic is an American artificial intelligence safety and research company founded in 2021 by former OpenAI researchers, among them the siblings Dario and Daniela Amodei. It develops the Claude family of large language models and ships them through a consumer assistant, an enterprise developer platform and the Claude Code agentic coding tool, alongside open standards such as the Model Context Protocol. Incorporated as a public benefit corporation and headquartered in San Francisco, the company concentrates on interpretability, alignment and reliability research and counts Google and Amazon among its largest investors.
anthropic.com • HQ: San Francisco, United States • Code Intelligence • AI Agents • Machine Learning • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2021
HQ: San Francisco, United States • Code Intelligence • AI Agents • Machine Learning • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2021
Top 25% payVerified live · 2 hours ago 5 months ago

Research Engineer, Machine Learning (RL Velocity)

$501k per year • In office (London, United Kingdom) • Visa sponsorship • Bachelor's Degree
JAX
Multimodal AI
PyTorch
Anthropic
Post-training
Pre-training
Apply
Open 342 daysTop 25% pay 12 months ago

Performance Engineer, GPU

$280k per year • In office (San Francisco, United States) • Visa sponsorship • Bachelor's Degree
Claude
CUDA Toolkit
Flash Attention
JAX
Multimodal AI
PyTorch
Quantization
CUDA
Triton
Anthropic
NCCL
NVLink
Apply
$500k per year • In office (San Francisco, United States) • Visa sponsorship • Bachelor's Degree
C++
Python
Rust
C++
PyTorch C++
TensorFlow C++
AI/ML
Claude
JAX
LLM
Multimodal AI
PyTorch
Reinforcement Learning
TensorFlow
Anthropic
Red Teaming
AI Agents
Function Calling
DevOps
Kubernetes
Apply
Report
JAX Jobs - Remote & On-site
Frequently asked questions
JAX: How many jobs are available now?
There are 491 JAX jobs listed on Alion right now.
JAX: What is the average salary for these roles?
The average salary for JAX roles on Alion is $285,074 USD.
JAX: Are remote, relocation, or visa sponsorship options available?
Many JAX listings allow remote or hybrid schedules, and some positions may offer relocation or visa sponsorship-verify on each job page.
JAX: How do I apply for JAX jobs on Alion?
Open the JAX job posting and click Apply or follow the employer’s link; sign in to save jobs and enable alerts for new listings.