Remote JAX Jobs

Remote JAX jobs hiring now at vetted startups and product companies on Alion, updated daily. Average salary around $222,164/year. Work from anywhere: fully remote positions with clear salary, time-zone and visa details. 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

Avride

avride.ai
avride.ai · HQ: Austin, United States · Ride Hailing · Delivery · Autonomous Driving · 501-1000 employees · Est. 2024 · Trust B · Hiring 43, 4 new
HQ: Austin, United States · Ride Hailing · Delivery · Autonomous Driving · 501-1000 employees · Est. 2024 · Trust B · Hiring 43, 4 new
Avride is an autonomous vehicle company that develops self-driving cars and sidewalk delivery robots, spun out of 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. The company is headquartered in Austin, Texas, United States.
No named decision-maker yet.
Careers mailbox is known
Open 442 daysVerified live · 1 hour ago 1 year ago Long shot · 7%

Machine Learning Engineer – Motion Planning & Prediction

≈ $134k – $299k per year (Estimated) · Austin · Remote (United States) · Relocation
Rust
C++
Python
C++
TensorFlow C++
AI/ML
JAX
NLP
TensorFlow
PyTorch
Machine Learning
AI Agents
Robotics
SLAM
Motion Planning
+4
Apply

Gradient

gradient.network
gradient.network · AI Compute & Inference · AI Agents · Trust B · Hiring 4
AI Compute & Inference · AI Agents · Trust B · Hiring 4
We are building the world’s first fully distributed AI runtime—a sovereign, peer-powered infrastructure where intelligence is hosted, served, and owned by the people.
No named decision-maker yet.
No careers mailbox known
Open 506 daysVerified live · 3 hours ago 1 year ago Long shot · 5%

AI Research Engineer

≈ $50k – $138k per year (Estimated) · Remote (Singapore) · Full-Time
Rust
Python
AI/ML
Fine-tuning
JAX
TensorFlow
Federated Learning
PyTorch
Machine Learning
Apply

Gensyn

gensyn.ai
gensyn.ai · HQ: London, United Kingdom · Blockchain Infrastructure · AI Compute & Inference · 51-200 employees · Est. 2020 · Startup · Hiring 1
HQ: London, United Kingdom · Blockchain Infrastructure · AI Compute & Inference · 51-200 employees · Est. 2020 · Startup · Hiring 1
Gensyn is a London company founded in 2020 that is building a protocol for distributed machine learning compute. Its network lets anyone contribute graphics processors to training jobs, with cryptographic proofs verifying that the work was performed correctly. The project is backed by a16z and aims to make training capacity a commodity market rather than the preserve of a few clouds.
No named decision-maker yet.
Careers mailbox is known
Open 561 daysVerified live · 1 hour ago Visa sponsorship 1 year ago Long shot · 3%

Research Intern

≈ $41k – $71k per year (Estimated) · Intern · Remote (likely Europe, CET hours) · Relocation · Master's Degree · Internship · Intern
JAX
TensorFlow
PyTorch
Machine Learning
Apply
Report

Anlatan

anlatan.ai
anlatan.ai · LLM & Generative AI · Trust B · Hiring 3
LLM & Generative AI · Trust B · Hiring 3
Anlatan is an AI company that makes the NovelAI storytelling product, working on cutting-edge AI with a focus on user-friendly, privacy-safe and censorship-free products. It describes itself as a small team of personalities and hires frontend and backend engineers, machine learning engineers and UI designers. The company is based in the United States.
No named decision-maker yet.
Careers mailbox is known
Open 1116 daysVerified live · 1 hour ago 3 years ago Long shot · 5%

Machine Learning Engineer

≈ $133k – $295k per year (Estimated) · Remote (United States) · Full-Time
Python
AI/ML
Fine-tuning
NLP
PyTorch
Machine Learning
JAX
Triton
DevOps
SLURM
Kubernetes
+1
Apply

Reka AI

reka.ai
reka.ai · HQ: Sunnyvale, United States · LLM & Generative AI · Multimodal AI · 201-500 employees · Est. 2022 · Startup · Trust B · Hiring 6
HQ: Sunnyvale, United States · LLM & Generative AI · Multimodal AI · 201-500 employees · Est. 2022 · Startup · Trust B · Hiring 6
Reka AI is a research laboratory building multimodal foundation models that process text, images, audio and video in one system. Its Core and Flash models were trained from scratch by a small team of researchers previously at DeepMind, Google Brain, Baidu and Meta, and they compete with far larger laboratories on multimodal benchmarks. The company sells its models to enterprises and governments that want capable multimodal systems without depending on the largest providers; its models are available through an API and through cloud marketplaces for enterprise deployment.
Dani Yogatama · Executive Officer, Director · decision-maker
Cyprien d'Autume · Director
Robert Toews · Director
Careers mailbox is known
Open 1192 daysVerified live · 4 hours ago Visa sponsorship 3 years ago Long shot · 8%

Member of Technical Staff (ML)

≈ $139k – $271k per year (Estimated) · Staff+ · Remote (United States, United Kingdom) · Full-Time · Staff
JAX
Multimodal AI
PyTorch
Machine Learning
Apply

AE Studio

ae.studio
ae.studio · HQ: Los Angeles, United States · AI Consulting & Integration · Custom Software Development · 11-50 employees · Est. 2015 · Trust B · Hiring 6, 1 new
HQ: Los Angeles, United States · AI Consulting & Integration · Custom Software Development · 11-50 employees · Est. 2015 · Trust B · Hiring 6, 1 new
AE Studio is a software development and data science consultancy that reinvests a share of its revenue into artificial intelligence alignment research rather than distributing it to outside shareholders. Its alignment team works on neglected approaches to making advanced models safe, including interpretability, attention schema theory and brain computer interface research, and publishes the results openly. The company funds that work through client engineering projects and an internal incubator that has produced several independent product ventures.
No named decision-maker yet.
Careers mailbox is known
Open 2033 daysVerified live · 3 hours ago 5 years ago Long shot · 5%

Applied AI Data Scientist

≈ $134k – $298k per year (Estimated) · Remote (United States, Brazil) · English: B2
Python
AI/ML
Fine-tuning
JAX
NLP
TensorFlow
Time Series
Computer Vision
Cursor
PyTorch
RAG
LLM
Machine Learning
Management
Kanban
Scrum
Agile
+7
Apply
Open 2033 daysVerified live · 3 hours ago 5 years ago Long shot · 5%

Applied AI Data Scientist (Brazil)

Florianópolis · Remote (Brazil) · English: B2
Python
AI/ML
Fine-tuning
JAX
NLP
TensorFlow
Time Series
Computer Vision
Cursor
PyTorch
RAG
LLM
Machine Learning
Management
Kanban
Scrum
Agile
+7
Apply
Machines can read this selection · checked 12:41 UTC
Remote JAX Jobs
Frequently asked questions
How many Remote JAX jobs are available on Alion?
There are currently 233 active listings from 90 hiring companies on Alion, refreshed daily.
What salary do Remote JAX jobs pay?
The average salary is around $222,164 per year based on the current listings.
Do Remote JAX jobs offer relocation or visa sponsorship?
Many listings also offer relocation packages or visa sponsorship - use the filters to narrow the list down to what you need.
How do I apply for Remote JAX jobs?
Open any listing on Alion and apply directly with the employer through the provided link - no lengthy sign-up required.