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

Pear VC

pear.vc
pear.vc • HQ: San Francisco, United States • Venture Capital • 51-200 employees • Est. 2013 • Trust A • Hiring 94, 2 new
HQ: San Francisco, United States • Venture Capital • 51-200 employees • Est. 2013 • Trust A • Hiring 94, 2 new
Pear VC is a venture capital firm specializing in pre-seed and seed-stage investments in startups. Founded by Pejman Nozad and Mar Hershenson, the firm provides hands-on support in areas such as product development, recruiting, fundraising, and go-to-market strategy. Pear has backed companies including DoorDash, Gusto, Aurora Solar, Vanta, and Guardant Health.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago 1 year ago Long shot · 15%

AI / ML Engineer - Known

≈ $174k – $328k per year (Estimated) • Middle • 3+ years exp • In office (San Francisco, United States) • Full-Time • Middle
Python
AI/ML
Embeddings
JAX
TensorFlow
PyTorch
LLM
Hugging Face
Human-in-the-Loop
LLM Guardrails
Agentic Workflows
Machine Learning
Reinforcement Learning
LLM Evaluation
Voice Agents
Analytics
A/B Testing
+7
Apply

METISINC

metisinc.io
metisinc.io • HQ: San Francisco, United States • IT Consulting & Digital Transformation • Custom Software Development • Est. 2025 • Trust B • Hiring 4
HQ: San Francisco, United States • IT Consulting & Digital Transformation • Custom Software Development • Est. 2025 • Trust B • Hiring 4
Asymmetrical Solutions Realisation & Asynchronous Delivery. Bespoke business design andengineering. Advisory, consulting, training, & recruitment.
No named decision-maker yet.
No careers mailbox known
Open 376 daysVerified live · 1 hour ago 1 year ago Long shot · 3%

Research Engineer

$200k per year • In office (San Francisco, United States)
Python
AI/ML
Reinforcement Learning
JAX
AI Agents
TensorFlow
PyTorch
Synthetic Data
OpenAI
Post-training
Reward Modeling
Machine Learning
DevOps
Platform Engineering
+4
Apply

Pinely

pinely.com
pinely.com • HQ: Amsterdam, Netherlands • Online Brokerage & Trading Platforms • Trust A • Hiring 17, 3 new
HQ: Amsterdam, Netherlands • Online Brokerage & Trading Platforms • Trust A • Hiring 17, 3 new
Pinely is a privately owned & funded algorithmic trading firm, based in Singapore, The Netherlands and Cyprus, and growing by leaps and bounds. We focus on high frequency and ultra low latency trading.
No named decision-maker yet.
Careers mailbox is known
Open 387 daysVerified live · 6 hours ago 1 year ago Long shot · 3%

ML Performance Engineer

≈ $61k – $129k per year (Estimated) • In office (Amsterdam, Netherlands) • Master's Degree • Full-Time
Python
AI/ML
CUDA Toolkit
JAX
Transformers
PyTorch
CUDA
Triton
Apply

Collate

collate.com
collate.com • HQ: San Francisco, United States • Clinical Trial Software • Document AI • 51-200 employees • Startup • Trust D • Hiring 11
HQ: San Francisco, United States • Clinical Trial Software • Document AI • 51-200 employees • Startup • Trust D • Hiring 11
Our mission Accelerate life-saving innovations by creating accurate documentation for the life sciences industry. Catalyze progress Automate paperwork at every stage of development and commercialization.
No named decision-maker yet.
No careers mailbox known
Likely inactiveVerified live · 2 hours ago 1 year ago Long shot · 4%

AI Research Scientist

≈ $163k – $358k per year (Estimated) • In office (San Francisco, United States)
JAX
NLP
TensorFlow
PyTorch
LLM
Machine Learning
Apply

Mantari

mantari.com
mantari.com • HQ: Torrance, United States • Geospatial AI • Energy AI • Startup • Trust B • Hiring 6
HQ: Torrance, United States • Geospatial AI • Energy AI • Startup • Trust B • Hiring 6
Earth, Quantified.. Weather drives advantage, delivering clarity for commerce, government, national security, and civilization itself.
No named decision-maker yet.
Careers mailbox is known
Open 401 daysVerified live · 3 hours ago 1 year ago Long shot · 3%

Software Engineer

≈ $120k – $253k per year (Estimated) • In office (Torrance, United States)
Python
Rust
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
CUDA Toolkit
JAX
TensorFlow
PyTorch
Ray
CUDA
DevOps
GCP
SLURM
Azure
CI/CD
AWS
Kubernetes
+9
Apply

Alinia

alinia.ai
alinia.ai • HQ: Dover, United States • AI Safety & Alignment • AI Agents • 11-50 employees • Est. 2023 • Startup • Hiring 2
HQ: Dover, United States • AI Safety & Alignment • AI Agents • 11-50 employees • Est. 2023 • Startup • Hiring 2
We are on a mission to make human-machine interactions safer in the most critical scenarios and at scale. Our core values At Alinia, we are all driven by a passion to make people’s lives better. We approach challenges with a growth mindset and strive to amplify others.
Ariadna Font Llitjos · Executive Officer, Director · decision-maker
Carlos Munoz Ferrandis · Executive Officer, Director · decision-maker
Manuel Silva Martinez · Director
Careers mailbox is known
Open 402 daysVerified live · 2 hours ago 1 year ago Long shot · 7%

Applied Research Scientist

≈ $48k – $112k per year (Estimated) • Hybrid (Barcelona, Spain) • Master's Degree • Full-Time
Python
AI/ML
JAX
NLP
TensorFlow
PyTorch
LLM
Data Augmentation
Synthetic Data
SFT
LLM Guardrails
Machine Learning
Post-training
+4
Apply

AI2C

ai2c.ch
ai2c.ch • Athens • Hiring 2
Athens • Hiring 2
The Science and Art of Continual Real-Time Learning. Computational Thinking Lab built by outstanding Minds.
No named decision-maker yet.
Careers mailbox is known
Open 404 daysVerified live · 3 hours ago 1 year ago Long shot · 5%

Senior Machine Learning Researcher

≈ $37k – $95k per year (Estimated) • Senior • In office (Athens, Greece) • PhD • Full-Time • Senior • English: B2
Python
AI/ML
Scikit-learn
JAX
TensorFlow
Keras
PyTorch
Time Series Forecasting
Machine Learning
Apply

Twenty

twenty.io
twenty.io • HQ: Arlington, United States • National Security • Cyber Warfare • 501-1000 employees • Est. 2024 • Trust A • Hiring 37, 4 new
HQ: Arlington, United States • National Security • Cyber Warfare • 501-1000 employees • Est. 2024 • Trust A • Hiring 37, 4 new
Twenty is a US defense technology company building cyber warfare capabilities to deter attacks on American and allied networks. It hires mission deployment leads and command mission operations leads for teams in Arlington, Chantilly, Fort Meade and New York. The company is headquartered in Arlington, Virginia, United States.
Joseph Lin · Executive Officer, Director · decision-maker
Larry Olson · Executive Officer · decision-maker
Raymond Tonsing · Director
No careers mailbox known
Top 25% payVerified live · 1 hour ago 1 year ago Long shot · 15%

Applied AI Engineer

$159k – $263k per year • Middle • 4+ years exp • In office (Arlington, United States) • Bachelor's Degree • Full-Time • Middle
Python
Databases
PostgreSQL
Weaviate
Milvus
pgvector
Pinecone
AI/ML
LangGraph
LangChain
RLHF
JAX
Pydantic AI
TensorFlow
PyTorch
CrewAI
LLM
RAG
DPO
SFT
PPO
GRPO
Post-training
Pre-training
vLLM
Quantization
Prompt Engineering
ONNX
TensorRT
AgentOps
LLMOps
DevOps
GCP
Azure
AWS
Docker
Kubernetes
+26
Apply

Datadog

datadoghq.com
datadoghq.com • HQ: New York, United States • Cloud Security • Log Management & Analytics • Observability & APM • 1001-5000 employees • Est. 2010 • NASDAQ: DDOG • Trust B • Hiring 411, 24 new
HQ: New York, United States • Cloud Security • Log Management & Analytics • Observability & APM • 1001-5000 employees • Est. 2010 • NASDAQ: DDOG • Trust B • Hiring 411, 24 new
Datadog is an American software company headquartered in New York that provides a cloud observability and security platform delivered as SaaS. Founded in 2010 and listed on Nasdaq, it covers infrastructure, application and log monitoring, security, digital experience, software delivery and AI observability, runs the DASH conference and has offices in Boston, Paris, Dublin, Amsterdam, Tokyo, Singapore and Sao Paulo. Openings span software and security engineering, sales and sales engineering, recruiting, enablement, finance, marketing and research roles in North America, Europe, Asia and Latin America.
Amit Agarwal · Executive Officer · decision-maker
Alexis Le-Quoc · Executive Officer, Director · decision-maker
Olivier Pomel · Executive Officer, Director · decision-maker
Chad Savoy · Executive Officer · decision-maker
Careers mailbox is known
Open 416 daysVerified live · 1 hour ago 1 year ago Long shot · 3%

AI Research Engineer - Datadog AI Research (DAIR)

≈ $68k – $144k per year (Estimated) • Equity • In office (Paris, France)
Python
Rust
C++
C++
PyTorch C++
AI/ML
Multimodal AI
Ray
Anomaly Detection
Time Series Forecasting
Post-training
World Models
DeepSpeed
CUDA Toolkit
Fine-tuning
RLHF
Quantization
JAX
PyTorch
CUDA
SFT
Megatron-LM
Speculative Decoding
DevOps
Datadog
SLURM
+15
Apply

Avride

avride.ai
avride.ai • HQ: Austin, United States • Ride Hailing • Delivery • Autonomous Driving • 501-1000 employees • Est. 2024 • Trust B • Hiring 74, 1 new
HQ: Austin, United States • Ride Hailing • Delivery • Autonomous Driving • 501-1000 employees • Est. 2024 • Trust B • Hiring 74, 1 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 423 daysVerified live · 1 hour ago 1 year ago Long shot · 5%

Machine Learning Engineer – Motion Planning & Prediction

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

Orbit

orbitneuro.com
orbitneuro.com • HQ: San Francisco, United States • Startup • Hiring 1
HQ: San Francisco, United States • Startup • Hiring 1
No named decision-maker yet.
No careers mailbox known
Open 430 daysTop 25% pay 1 year ago Long shot · 4%

Founding Machine Learning Engineer

$225k – $275k per year • Middle • 3+ years exp • In office (San Francisco, United States) • Bachelor's Degree • Full-Time • Middle
Python
AI/ML
JAX
Multimodal AI
TensorFlow
PyTorch
Self-Supervised Learning
Time Series Forecasting
Hugging Face
Machine Learning
Management
Dropbox
+2
Apply
Report

Oligo Space

oligospace.com
oligospace.com • HQ: Hawthorne, United States • Satellite Manufacturing • Space Services • 1-10 employees • Est. 2024 • Startup • Trust B • Hiring 4
HQ: Hawthorne, United States • Satellite Manufacturing • Space Services • 1-10 employees • Est. 2024 • Startup • Trust B • Hiring 4
Oligo Space is a space company building a manufacturing-in-the-loop foundation model to automate spacecraft design and production. It handles design, manufacturing, launch and operations for customers and hires spacecraft and AI engineers. The company is headquartered in Hawthorne, California, United States.
No named decision-maker yet.
No careers mailbox known
Open 440 daysTop 25% pay 1 year ago Long shot · 2%

AI Space Systems Software Engineer

$110k – $164k per year • Junior • 2+ years exp • In office (Hawthorne, United States) • Bachelor's Degree • Full-Time • Junior
Python
Python
FastAPI
AI/ML
LangChain
Reinforcement Learning
JAX
Multimodal AI
AI Agents
TensorFlow
PyTorch
LLM
RAG
Ray
CLIP
Knowledge Graph
Edge AI
DevOps
Docker
Robotics
Reinforcement Learning
+9
Apply

Pulse

runpulse.com
runpulse.com • HQ: San Francisco, United States • Information Security • Computer Vision • Cybersecurity • Document AI • 201-500 employees • Est. 2024 • Startup • Trust D • Hiring 10
HQ: San Francisco, United States • Information Security • Computer Vision • Cybersecurity • Document AI • 201-500 employees • Est. 2024 • Startup • Trust D • Hiring 10
Pulse converts complex PDFs, scans and spreadsheets into verified structured data for enterprise workflows. It targets documents that defeat ordinary OCR, such as financial statements and dense tables. Customers use the output to feed downstream systems without manual keying.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveVerified live · 2 hours ago 1 year ago Long shot · 3%

Machine Learning Engineer

$160k – $220k per year • In office (San Francisco, United States) • Relocation • Full-Time
Computer Vision
NLP
OCR
Triton Inference Server
Quantization
JAX
Multimodal AI
ONNX
TensorRT
PyTorch
+2
Apply

Autonomous Technologies Group

atg-intelligence.com
atg-intelligence.com • HQ: New York, United States • AI Agents • Defense AI • Trust B • Hiring 7
HQ: New York, United States • AI Agents • Defense AI • Trust B • Hiring 7
Autonomous Technologies Group is an applied AI research group developing high-performance reasoning systems to operate on problems of monumental scale and complexity.
No named decision-maker yet.
No careers mailbox known
Open 459 daysVerified live · 1 hour ago 1 year ago Long shot · 9%

Member of Technical Staff (Research Engineering)

$180k – $260k per year • Staff+ • Hybrid (New York, San Francisco, United States) • Full-Time • Staff
Python
AI/ML
Cursor
Claude Code
Reinforcement Learning
JAX
AI Agents
TensorFlow
PyTorch
OpenAI Codex
Management
Stripe
+2
Apply
Open 459 daysTop 25% pay 1 year ago Long shot · 3%

Research Scientist

$200k – $300k per year • Hybrid (New York, San Francisco, United States) • Full-Time
Python
AI/ML
Cursor
Claude Code
Reinforcement Learning
JAX
PyTorch
OpenAI Codex
AI Agents
Management
Stripe
+1
Apply

Boson AI

boson.ai
boson.ai • HQ: Santa Clara, United States • HQ: Toronto, Canada • Speech & Audio AI • 11-50 employees • Est. 2023 • Likely startup • Trust B • Hiring 7
HQ: Santa Clara, United States • HQ: Toronto, Canada • Speech & Audio AI • 11-50 employees • Est. 2023 • Likely startup • Trust B • Hiring 7
Boson AI builds voice agents for business-critical work — speech-to-speech models, real-time avatars, and production TTS in 100+ languages.
No named decision-maker yet.
Careers mailbox is known
Open 487 daysTop 25% pay 1 year ago Long shot · 7%

Machine Learning Engineer - Enterprise

$106k – $283k per year • In office (Toronto, Canada) • Bachelor's Degree • Full-Time
Python
Rust
TypeScript
AI/ML
Fine-tuning
JAX
Multimodal AI
AI Agents
PyTorch
LLM
RAG
Edge AI
Machine Learning
DevOps
GitHub
GCP
WebRTC
Azure
AWS
+9
Apply

Etched

etched.com
etched.com • HQ: San Jose, United States • AI Compute & Inference • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2022 • Trust A • Hiring 79, 10 new
HQ: San Jose, United States • AI Compute & Inference • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2022 • Trust A • Hiring 79, 10 new
Etched is an American semiconductor company founded in 2022 that designs application-specific chips for running transformer-based AI models, together with the servers, racks, software and manufacturing methods around them. Its Sohu chip targets high throughput, low latency and lower cost and power for inference of frontier models, and the company has raised large venture rounds to bring it to production, with teams in San Jose, Austin and Taiwan. The company is headquartered in San Jose, California, United States.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveTop 25% pay Visa sponsor 1 year ago Long shot · 8%

Inference Software Engineer

$175k – $400k per year • In office (San Jose, United States) • Relocation • Full-Time
Rust
C++
C++
PyTorch C++
AI/ML
JAX
PyTorch
TPU
InfiniBand
NVLink
Transformers
Mixture of Experts
DevOps
Linux
+3
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 487 daysVerified live · 2 hours ago 1 year ago Long shot · 5%

AI Research Engineer

Remote (Singapore) • Full-Time
Python
Rust
AI/ML
Fine-tuning
JAX
AI Agents
TensorFlow
PyTorch
Federated Learning
Machine Learning
Management
Discord
Telegram
+3
Apply

Point72

point72.com
point72.com • HQ: Stamford, United States • Venture Capital • Hedge Funds & Alternative Investments • 1001-5000 employees • Est. 2014 • Trust B • Hiring 182, 7 new
HQ: Stamford, United States • Venture Capital • Hedge Funds & Alternative Investments • 1001-5000 employees • Est. 2014 • Trust B • Hiring 182, 7 new
Point72 is an American investment firm founded in 2014 when Steven Cohen converted his former hedge fund SAC Capital into a family office, before reopening to outside capital in 2018. It runs a multi-strategy business dominated by discretionary long and short equity teams organised as independent portfolio managers, alongside macro, systematic and quantitative strategies that rely heavily on alternative data and machine learning. Headquartered in Stamford, Connecticut with offices across North America, Europe and Asia, it also operates Point72 Ventures as a venture capital arm and runs a well-known academy programme that trains analysts from outside finance.
Steven Cohen · Founder, Chairman and Chief Executive Officer · decision-maker
Harry Schwefel · Co-Chief Investment Officer · decision-maker
Careers mailbox is known
Likely inactiveVerified live · 1 hour ago Visa sponsor 1 year ago Long shot · 5%

Quantitative Researcher - Machine Learning

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

LawZero

lawzero.org
lawzero.org • HQ: Montreal, Canada • AI Safety & Alignment • 201-500 employees • Est. 2025 • Trust B • Hiring 11
HQ: Montreal, Canada • AI Safety & Alignment • 201-500 employees • Est. 2025 • Trust B • Hiring 11
LawZero is a non-profit laboratory building non agentic and verifiable artificial intelligence systems. It was founded by the deep learning pioneer Yoshua Bengio. The organisation researches guardrail models that supervise other systems.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveVerified live · 2 hours ago 1 year ago Long shot · 8%

Senior ML Research Scientist

≈ $77k – $157k per year (Estimated) • Senior • 5+ years exp • In office (Berlin, Germany, Montreal, Canada) • PhD • Senior
Python
AI/ML
Machine Learning
Fine-tuning
JAX
AI Agents
Transfer Learning
TensorFlow
PyTorch
Apply
Likely inactiveVerified live · 2 hours ago 1 year ago Long shot · 8%

Senior ML Research Developer

≈ $76k – $155k per year (Estimated) • Senior • 5+ years exp • In office (Berlin, Germany, Montreal, Canada) • Senior
JAX
AI Agents
TensorFlow
PyTorch
Machine Learning
DevOps
GCP
Azure
AWS
Docker
Kubernetes
HPC
+3
Apply

LoiZéro

loizero.ai
loizero.ai • HQ: Montreal, Canada • AI Research Labs • AI Safety & Alignment • Trust B • Hiring 8
HQ: Montreal, Canada • AI Research Labs • AI Safety & Alignment • Trust B • Hiring 8
LoiZéro is a non-profit organization that advances research and develops technical solutions for designing safe AI systems, based on new methods proposed by Professor Yoshua Bengio. Its research aims at non-agentic AI systems, and it hires machine learning researchers and engineers. The organization is based in Montreal, Canada.
No named decision-maker yet.
No careers mailbox known
Open 501 daysVerified live · 29 min ago 1 year ago Long shot · 5%

Chercheur.se en apprentissage automatique sénior

≈ $77k – $159k per year (Estimated) • Senior • 5+ years exp • In office (Berlin, Germany, Montreal, Canada) • PhD • Senior • French: C1
Python
AI/ML
JAX
TensorFlow
PyTorch
Apply
Open 501 daysVerified live · 29 min ago 1 year ago Long shot · 5%

Développeur.se de recherche en apprentissage automatique sénior

≈ $76k – $155k per year (Estimated) • Senior • 5+ years exp • In office (Berlin, Germany, Montreal, Canada) • Senior • French: C1
JAX
TensorFlow
PyTorch
DevOps
GCP
Azure
AWS
Docker
Kubernetes
Apply
Machines can read this selection · checked 10:47 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.