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

Optiver

optiver.com
optiver.com • HQ: Amsterdam, Netherlands • Hedge Funds & Alternative Investments • 1001-5000 employees • Est. 1986 • Trust B • Hiring 145, 3 new
HQ: Amsterdam, Netherlands • Hedge Funds & Alternative Investments • 1001-5000 employees • Est. 1986 • Trust B • Hiring 145, 3 new
Optiver is a Dutch proprietary trading firm founded in 1986 and headquartered in Amsterdam. It builds ultra-low-latency trading systems and hardware to provide liquidity on exchanges worldwide. The company is one of the largest technology employers in Amsterdam despite trading only its own capital.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago Visa sponsor 17 days ago Hiring now · 87%

AI Lab - Research

$200k per year • In office (New York, United States)
Python
AI/ML
CUDA Toolkit
Reinforcement Learning
JAX
AI Agents
Mamba
Transformers
PyTorch
Self-Supervised Learning
Time Series Forecasting
CUDA
Triton
Post-training
Pre-training
Machine Learning
+7
Apply
Verified live · 1 hour ago 2 months ago Fading · 29%

2027 Shanghai Machine Learning Engineer Summer Internship

Intern • In office (Shanghai, China) • Internship • Intern
JAX
PyTorch
Machine Learning
MLFlow
Apply
Likely inactiveVerified live · 1 hour ago Visa sponsor 8 months ago Long shot · 5%

Machine Learning Research Engineer

$200k per year • In office (New York, United States) • PhD
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
CUDA Toolkit
Flax
CUDA
Triton
XLA
+5
Apply

Xanadu

xanadu.ai
xanadu.ai • HQ: Toronto, Canada • Quantum Computing Hardware • Est. 2016 • Hiring 82, 7 new
HQ: Toronto, Canada • Quantum Computing Hardware • Est. 2016 • Hiring 82, 7 new
Xanadu (Xanadu Quantum Technologies Inc.) is a prominent photonic quantum computing enterprise headquartered in Toronto, Canada. Founded in 2016 by CEO Christian Weedbrook, the venture has raised over $250 million from prominent investors including Bessemer Venture Partners, Capricorn Investment Group, and Real Ventures, alongside major government backing from Canada and DARPA.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 17 days ago Likely open · 69%

Quantum Scientist - Software

≈ $66k – $194k per year (Estimated) • In office (Toronto, Canada) • Master's Degree • Full-Time
Python
AI/ML
JAX
SciPy
NumPy
Quantum
PennyLane
Apply
Verified live · 2 hours ago 1 month ago Fading · 29%

Quantum Scientist — Finance

≈ $65k – $193k per year (Estimated) • Hybrid (Toronto, Canada) • PhD • Full-Time
Python
AI/ML
Time Series Forecasting
Machine Learning
JAX
PyTorch
Quantum
PennyLane
Apply
Verified live · 2 hours ago 2 months ago Long shot · 14%

Quantum Scientist - Compilation

≈ $63k – $186k per year (Estimated) • Hybrid (Toronto, Canada) • Bachelor's Degree • Full-Time
Python
C++
C++
LLVM
AI/ML
JAX
MLIR
Machine Learning
Quantum
PennyLane
Apply

Titan OS

titanos.tv
titanos.tv • HQ: Barcelona, Spain • Streaming & OTT Platforms • Operating Systems • 201-500 employees • Startup • Hiring 24, 2 new
HQ: Barcelona, Spain • Streaming & OTT Platforms • Operating Systems • 201-500 employees • Startup • Hiring 24, 2 new
Titan OS, independent operating system for CTV, offering a collaborative business model with revenue sharing and UX customisation for manufacturers and monetisation and data support for content owners.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 17 days ago Fading · 37%

AI Engineer

≈ $46k – $106k per year (Estimated) • In office (Barcelona, Spain) • Bachelor's Degree • Full-Time
Python
Go
SQL
Ruby
JavaScript
Node JS
Databases
PostgreSQL
Google BigQuery
BigQuery
Apache Kafka
AI/ML
Spark
JAX
Function Calling
TensorFlow
PyTorch
LLM
RAG
Recommender Systems
Machine Learning
Embeddings
Flink
DevOps
Rest API
GitHub Actions
CI/CD
Docker
GCP
Prometheus
AWS
Grafana
Analytics
A/B Testing
Management
Agile
+23
Apply

Akrys.ai

akrys.ai
akrys.ai • HQ: Bengaluru, India • Document AI • Speech & Audio AI • EHR & Clinical Software • Medical AI • Est. 2025 • Likely startup • Hiring 5
HQ: Bengaluru, India • Document AI • Speech & Audio AI • EHR & Clinical Software • Medical AI • Est. 2025 • Likely startup • Hiring 5
Akrys AI is a healthtech company building a clinical intelligence platform to convert doctor-patient conversations into structured records and clinical notes, offering AI-assisted support for healthcare workflows.
No named decision-maker yet.
Careers mailbox is known
17 days ago Likely open · 61%

Lead AI Engineer - Speech & Clinical AI

≈ $25k – $63k per year (Estimated) • Lead • 6+ years exp • In office (Bengaluru, India) • Staff
Python
SQL
Bash
Rust
C++
Python
Dask
C++
PyTorch C++
TensorFlow C++
AI/ML
Weights & Biases
Qwen
Spark
Bitsandbytes
DeepSpeed
LoRA
vLLM
MLFlow
Triton Inference Server
Gemma
Fine-tuning
RLHF
Quantization
Scikit-learn
Prompt Engineering
SciPy
Knowledge Distillation
AI Agents
NLP
NER
Speech Recognition
AWQ
GPTQ
ONNX
TensorRT
PEFT
QLoRA
TGI
TRL
Llama
Mistral
Accelerate
Kaldi
Transformers
NumPy
PyTorch
LLM
RAG
Whisper
Ray
Hallucination
Synthetic Data
Encoder-Decoder
Hugging Face
DPO
SFT
PPO
Pre-training
FSDP
NVIDIA NeMo
TorchServe
NCCL
Model Distillation
Flash Attention
JAX
xFormers
TensorFlow
DevOps
GCP
SLURM
Azure
CI/CD
Git
AWS
Docker
Kubernetes
SOAP
Cybersecurity
GDPR
HIPAA
+68
Apply

Vantor

vantor.com
vantor.com • HQ: Westminster, United States • Space Services • Geospatial Analytics • Earth Observation • 201-500 employees • Est. 2025 • Trust B • Hiring 145, 16 new
HQ: Westminster, United States • Space Services • Geospatial Analytics • Earth Observation • 201-500 employees • Est. 2025 • Trust B • Hiring 145, 16 new
Vantor is an American space intelligence company created in 2025 when Maxar Technologies split its Earth imaging business from its satellite manufacturing arm and renamed it. It operates one of the largest commercial high-resolution imaging constellations, sells satellite imagery, three-dimensional terrain data and analytics to defence, intelligence and government customers, and holds long-running United States government imagery contracts. Headquartered in Westminster, Colorado and owned by Advent International, it competes with the newer smallsat operators while retaining the archive and geospatial accuracy built up over decades.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 18 days ago Likely open · 57%

Principal Applied AI Scientist

$150k – $200k per year • Lead • 8+ years exp • United States • Remote (United States) • Bachelor's Degree • Full-Time • Principal
Python
AI/ML
JAX
Multimodal AI
Computer Vision
TensorFlow
PyTorch
Machine Learning
Fine-tuning
AI Agents
DevOps
GCP
+2
Apply
Verified live · 2 hours ago 2 months ago Long shot · 22%

Applied AI Scientist

$128k – $170k per year • Senior • 8+ years exp • United States • Remote (United States) • Bachelor's Degree • Full-Time • Staff
Python
AI/ML
JAX
Multimodal AI
Computer Vision
TensorFlow
PyTorch
Machine Learning
Fine-tuning
AI Agents
DevOps
GCP
+2
Apply

Amgen

amgen.com
amgen.com • HQ: Thousand Oaks, United States • Immunology & Inflammation Therapeutics • Cardiometabolic Therapeutics • Prescription Drugs • Biologics & Biosimilars • 5000+ employees • Est. 1980 • Trust B • Hiring 1273, 92 new
HQ: Thousand Oaks, United States • Immunology & Inflammation Therapeutics • Cardiometabolic Therapeutics • Prescription Drugs • Biologics & Biosimilars • 5000+ employees • Est. 1980 • Trust B • Hiring 1273, 92 new
Amgen is an American biotechnology company founded in Thousand Oaks in 1980 that became one of the first firms to turn recombinant protein research into large commercial medicines. Its history rests on Epogen and Neupogen, and its current portfolio spans bone health with Prolia and Evenity, inflammation with Otezla and Tezspire, cardiovascular disease with Repatha, oncology including the bispecific antibody Blincyto, and a biosimilars business built on its own manufacturing capacity. The company is listed on Nasdaq, is a component of the Dow Jones Industrial Average, and expanded further into rare disease through its acquisition of Horizon Therapeutics.
Jay Bradner · Chief Scientific Officer · decision-maker
Robert Bradway · Chairman and Chief Executive Officer · decision-maker
George Rathmann · Founding Chief Executive Officer · founder
Careers mailbox is known
Verified live · 1 hour ago 18 days ago Likely open · 50%

Sr. Associate - Machine Learning Engineer

≈ $27k – $60k per year (Estimated) • Senior • 5+ years exp • In office (Hyderabad, India) • Master's Degree • Full-Time • Senior
Python
AI/ML
Fine-tuning
Embeddings
JAX
AI Agents
NLP
TensorFlow
PyTorch
LLM
Human-in-the-Loop
LLM Guardrails
Interpretability
Tool Use
Machine Learning
Multi-Agent Systems
DevOps
CI/CD
Kubernetes
+9
Apply
Top 25% payVerified live · 1 hour ago Visa sponsor 22 days ago Fading · 43%

Director, AI & Machine Learning

$243k – $328k per year • Head+ • 12+ years exp • Thousand Oaks • Remote (United States) • Bachelor's Degree • Full-Time • Architect
Python
Databases
Databricks
Snowflake
AI/ML
JAX
AI Agents
TensorFlow
PyTorch
Machine Learning
Function Calling
LLM Guardrails
Tool Use
DevOps
GCP
Azure
CI/CD
AWS
+7
Apply
Verified live · 1 hour ago 1 month ago Fading · 26%

Senior Data Scientist - Protein Structure ML models

≈ $27k – $55k per year (Estimated) • Senior • 6+ years exp • In office (Hyderabad, India) • Bachelor's Degree • Full-Time • Senior
Python
AI/ML
Machine Learning
JAX
PyTorch
Apply

Merck & Co.

msd.com
msd.com • HQ: Rahway, United States • Biologics & Biosimilars • Veterinary Medicine • Vaccines • Oncology Therapeutics • Prescription Drugs • 5000+ employees • Est. 1891 • Trust A • Hiring 1286, 167 new
HQ: Rahway, United States • Biologics & Biosimilars • Veterinary Medicine • Vaccines • Oncology Therapeutics • Prescription Drugs • 5000+ employees • Est. 1891 • Trust A • Hiring 1286, 167 new
Merck & Co. is an American pharmaceutical company founded in 1891 as the United States arm of the German firm Merck and made independent after the First World War, trading as MSD outside the United States and Canada. Its portfolio is dominated by the immuno-oncology drug Keytruda, alongside vaccines such as Gardasil and Vaxneuvance, hospital acute care products, and a large animal health business covering livestock and companion animals. The company is headquartered in Rahway, New Jersey, listed on the New York Stock Exchange, and spends a very large share of revenue on research into oncology, infectious disease and cardiometabolic therapies.
Dean Li · President, Merck Research Laboratories · decision-maker
Robert Davis · Chairman and Chief Executive Officer · decision-maker
George Merck · Founder · founder
Careers mailbox is known
Verified live · 1 hour ago Visa sponsorship 18 days ago Fading · 27%

Associate Scientist, Post Doc Fellow- AI/ML & Computational Biology for Antigen Design

$82k – $92k per year • Junior • In office (West Point, United States) • Relocation • PhD • Full-Time • Principal
Python
AI/ML
JAX
PyTorch
Machine Learning
DevOps
HPC
Apply

GUARDIARIS

guardiaris.com
guardiaris.com • Ljubljana • Small Arms • Military Training • Military • 51-200 employees • Hiring 5, 3 new
Ljubljana • Small Arms • Military Training • Military • 51-200 employees • Hiring 5, 3 new
Military training simulators redefined. Guardiaris builds custom solutions for small arms, anti-tank weapons, vehicles and drone operator training.
No named decision-maker yet.
No careers mailbox known
In office (Ljubljana, Slovenia) • Full-Time
Rust
C++
Bash
C++
TensorFlow C++
PyTorch C++
AI/ML
OpenCV
Fine-tuning
Scikit-learn
JAX
SciPy
Transfer Learning
TensorFlow
NumPy
PyTorch
Data Augmentation
DevOps
Linux
Analytics
Matplotlib
+9
Apply

Anthropic

anthropic.com
anthropic.com • HQ: San Francisco, United States • Code Intelligence • AI Research Labs • AI Safety & Alignment • Foundation Models • LLM & Generative AI • 1001-5000 employees • Est. 2021 • Trust B • Hiring 648, 39 new
HQ: San Francisco, United States • Code Intelligence • AI Research Labs • AI Safety & Alignment • Foundation Models • LLM & Generative AI • 1001-5000 employees • Est. 2021 • Trust B • Hiring 648, 39 new
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.
Jared Kaplan · Co-Founder & Chief Science Officer · decision-maker
Daniela Amodei · Co-Founder & President · decision-maker
Dario Amodei · Co-Founder & Chief Executive Officer · decision-maker
Tom Brown · Co-Founder · founder
No careers mailbox known
Top 25% payVerified live · 2 hours ago 18 days ago Likely open · 50%

Research Engineer / Research Scientist, RL Frontiers

$500k per year • Hybrid (San Francisco, United States) • Bachelor's Degree
Python
Rust
C++
C++
PyTorch C++
AI/ML
Claude
Reinforcement Learning
JAX
PyTorch
Anthropic
Multimodal AI
TPU
Interpretability
+4
Apply
Top 25% payVerified live · 2 hours ago 1 month ago Fading · 44%

Staff+ Software Engineer, ML Inference Path

$320k – $485k per year • Staff+ • 5+ years exp • Hybrid (San Francisco, United States) • Bachelor's Degree • Staff
Python
AI/ML
Claude
JAX
Multimodal AI
TensorFlow
PyTorch
Anthropic
Interpretability
Machine Learning
Analytics
A/B Testing
+2
Apply
Open 176 daysTop 25% pay 6 months ago Long shot · 4%

Research Engineer, Machine Learning (RL Velocity)

$493k – $840k per year • Hybrid (London, United Kingdom) • Bachelor's Degree
JAX
Multimodal AI
PyTorch
Anthropic
Post-training
Pre-training
Interpretability
Apply

Universalagi

universalagi.com
universalagi.com • HQ: San Francisco, United States • Machine Learning • AI for Science • Simulation & Digital Twin Software • 201-500 employees • Est. 2025 • Likely startup • Hiring 19, 1 new
HQ: San Francisco, United States • Machine Learning • AI for Science • Simulation & Digital Twin Software • 201-500 employees • Est. 2025 • Likely startup • Hiring 19, 1 new
The platform for engineering teams building and operating Physics AI at scale. Predict aerodynamic performance, structural loads, and thermal behavior in seconds.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago Visa sponsor 18 days ago Likely open · 65%

CAD Automation Researcher

≈ $165k – $361k per year (Estimated) • In office (San Francisco, United States) • PhD • Full-Time
Python
Databases
Databricks
AI/ML
Embeddings
Reinforcement Learning
JAX
Diffusion Models
PyTorch
Apply

Oracle

oracle.com
oracle.com • HQ: Austin, United States • CRM Software • HR & Payroll Software • ERP & Business Management Software • Cloud Platforms (IaaS & PaaS) • Databases & DBMS • 5000+ employees • Est. 1977 • NYSE: ORCL • Trust A • Hiring 1133, 105 new
HQ: Austin, United States • CRM Software • HR & Payroll Software • ERP & Business Management Software • Cloud Platforms (IaaS & PaaS) • Databases & DBMS • 5000+ employees • Est. 1977 • NYSE: ORCL • Trust A • Hiring 1133, 105 new
Oracle is an American enterprise technology company founded in 1977 by Larry Ellison, Bob Miner and Ed Oates, and headquartered in Austin, Texas. It built its business on the Oracle Database, still the reference relational engine for large transactional systems, and has expanded into a full applications suite covering finance, human resources, supply chain and customer experience through Fusion Cloud and NetSuite. Its fastest growing segment is Oracle Cloud Infrastructure, which the company has positioned aggressively for AI training and inference workloads through large multi-year capacity contracts.
Larry Ellison · Co-Founder, Chairman & Chief Technology Officer · decision-maker
Clay Magouyrk · Chief Executive Officer · decision-maker
Safra Catz · Executive Vice Chair · decision-maker
Ed Oates · Co-Founder · founder
No careers mailbox known
Top 25% payVerified live · 5 min ago Visa sponsor 18 days ago Long shot · 13%

Senior Principal Engineer - AI Networking

$135k – $306k per year • Equity • Lead • 10+ years exp • In office (Seattle, United States) • Bachelor's Degree • Principal
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
InfiniBand
DeepSpeed
JAX
TensorFlow
PyTorch
Megatron-LM
NCCL
DevOps
HPC
Linux
Kubernetes
+5
Apply

Laminar

runlaminar.com
runlaminar.com • HQ: Boston, United States • Food & Beverages • Sensors & Transducers • Industrial AI • 11-50 employees • Startup • Trust B • Hiring 16
HQ: Boston, United States • Food & Beverages • Sensors & Transducers • Industrial AI • 11-50 employees • Startup • Trust B • Hiring 16
Laminar uses science-led AI and in-line spectral sensors to optimize clean-in-place, product changeovers, and line performance for food, beverage, and CPG manufacturers.
David Lu · Executive Officer, Director · decision-maker
Annie Lu · Executive Officer, Director · decision-maker
Dayna Grayson · Director
Jim Moffatt · Director
Careers mailbox is known
Verified live · 15 min ago Visa sponsor 18 days ago Likely open · 53%

(Jr) Machine Learning (ML) Developer

$89k – $141k per year • Junior • Hybrid (Boston, United States) • Full-Time • Junior
Python
Databases
Databricks
AI/ML
Polars
Scikit-learn
JAX
Multimodal AI
TensorFlow
Pandas
NumPy
PyTorch
Self-Supervised Learning
Time Series Forecasting
Machine Learning
DevOps
GCP
AWS
+7
Apply

Quest Global

questglobal.com
questglobal.com • HQ: Bengaluru, India • Engineering Services • Est. 1997 • Hiring 20, 3 new
HQ: Bengaluru, India • Engineering Services • Est. 1997 • Hiring 20, 3 new
Quest Global provides engineering research and development services to global industries. Its work spans aerospace, medical devices, energy, rail and semiconductors. The company operates delivery centres across many countries.
No named decision-maker yet.
Careers mailbox is known
19 days ago Likely open · 48%

Full Stack developer

≈ $15k – $35k per year (Estimated) • Senior • 6+ years exp • In office (Pune, India) • Full-Time
Python
JavaScript
SQL
Java
Java
Spring Framework
Hibernate
Databases
Oracle
DynamoDB
AI/ML
Copilot
Claude Code
JAX
AI Agents
AWS Bedrock
LLM
RAG
OpenAI Codex
Frontend
React.js
DevOps
Rest API
Azure
CI/CD
AWS
AWS Lambda
Amazon S3
SOAP
Management
Agile
QA
Selenium
Playwright
Rest-Assured
k6
+21
Apply

Somapa IT

somapait.com
somapait.com • Bangkok • Custom Software Development • IT Consulting & Digital Transformation • Est. 1998 • Hiring 5
Bangkok • Custom Software Development • IT Consulting & Digital Transformation • Est. 1998 • Hiring 5
At Somapa Information Technology, our commitment to excellence and innovation is embodied in our flagship initiative, Smart Border Solutions.
No named decision-maker yet.
Careers mailbox is known
19 days ago Likely open · 47%

Full Stack Web Developer (Urgently Required)

≈ $9.5k – $24k per year (Estimated) • In office (Bangkok, Thailand) • Full-Time
Python
JavaScript
PHP
SQL
C#
C++
Java
TypeScript
Java
Spring Boot
C#
.NET
Databases
MySQL
Oracle
MariaDB
AI/ML
JAX
Frontend
Angular
React.js
JQuery
DevOps
Azure
CI/CD
Windows Server
AWS
Docker
Kubernetes
Linux
+16
Apply
22 days ago Likely open · 47%

Full Stack Web Developer

$11k – $18k per year (gross) • In office (Bangkok, Thailand) • Full-Time
Java
SQL
JavaScript
TypeScript
Java
Spring Boot
Databases
MySQL
Oracle
MariaDB
AI/ML
JAX
Frontend
Angular
React.js
DevOps
Azure
CI/CD
Windows Server
AWS
Docker
Kubernetes
Linux
+10
Apply

Southampton

southampton.ac.uk
southampton.ac.uk • Southampton • Higher Education • 11-50 employees • Hiring 6
Southampton • Higher Education • 11-50 employees • Hiring 6
Get to know more about the University of Southampton. Explore our history, mission, and values. Learn about our commitment to education, research and community.
No named decision-maker yet.
No careers mailbox known
$55k – $60k per year • Southampton • Remote (likely United Kingdom, UK time hours) • Full-Time
Python
C++
MATLAB
C++
TensorFlow C++
PyTorch C++
AI/ML
Machine Learning
JAX
TensorFlow
PyTorch
+1
Apply
Hybrid (Southampton, United Kingdom) • Master's Degree • Contractor
Python
C++
MATLAB
C++
TensorFlow C++
PyTorch C++
AI/ML
Machine Learning
JAX
TensorFlow
PyTorch
+1
Apply
lseg.com • HQ: London, United Kingdom • Data Providers & Datasets • Financial Data & Market Intelligence • Stock Exchanges & Market Infrastructure • 5000+ employees • Est. 2007 • Trust B • Hiring 145, 2 new
HQ: London, United Kingdom • Data Providers & Datasets • Financial Data & Market Intelligence • Stock Exchanges & Market Infrastructure • 5000+ employees • Est. 2007 • Trust B • Hiring 145, 2 new
London Stock Exchange Group (LSEG) is a global financial markets infrastructure and data provider headquartered in London, England. Formed as a group in 2007, it owns the London Stock Exchange, LSEG Data and Analytics built on the former Refinitiv, FTSE Russell and LCH, and holds a majority stake in Tradeweb, with large operations centres in Bengaluru, Colombo, Bucharest, Gdynia and Taguig. It hires software and data engineers, site reliability engineers, valuations and post-trade specialists, customer success and go-to-market managers, finance staff and journalists for services such as ZAWYA.
No named decision-maker yet.
Careers mailbox is known
Verified live · 11 hours ago 19 days ago Long shot · 20%

Enterprise Data Scientist

≈ $61k – $147k per year (Estimated) • Staff+ • 10+ years exp • In office (Mumbai, India) • Bachelor's Degree • Full-Time • Staff
Python
SQL
Python
Dask
Databases
Weaviate
Cassandra
Pinecone
DynamoDB
ElasticSearch
Apache Kafka
AI/ML
AI Agents
NLP
RAG
LLMOps
LLM Guardrails
Machine Learning
LangGraph
AutoGen
LangChain
Claude
Qwen
Spark
Airflow
DeepSeek
LoRA
Model Context Protocol
Fine-tuning
Reinforcement Learning
Quantization
Scikit-learn
JAX
SciPy
Multimodal AI
Knowledge Distillation
Function Calling
Transfer Learning
Arize Phoenix
LangSmith
PEFT
QLoRA
spaCy
Llama
Transformers
TensorFlow
Pandas
NumPy
PyTorch
CrewAI
Gemini
LLM
BERT
NLTK
Ray
Hallucination
Semantic Search
Hybrid Search
Sentiment Analysis
Hugging Face
GPT-5
Semantic Search
EU AI Act
Multi-Agent Systems
Model Distillation
DevOps
GCP
Azure
AWS
Analytics
Tableau
Power BI
Plotly
+60
Apply

PostEra

postera.ai
postera.ai • HQ: Cambridge, United States • Drug Discovery AI • Drug Discovery • 51-200 employees • Est. 2019 • Startup • Hiring 1
HQ: Cambridge, United States • Drug Discovery AI • Drug Discovery • 51-200 employees • Est. 2019 • Startup • Hiring 1
PostEra uses machine learning for medicinal chemistry, proposing synthesis routes and analogues for drug discovery programmes. Founded in 2019 by Cambridge chemists, it coordinated the open science COVID Moonshot antiviral project. The company partners with major pharmaceutical firms on generative chemistry.
Aaron Morris · Executive Officer, Director · decision-maker
Alpha Lee · Executive Officer, Director · decision-maker
Matthew Robinson · Executive Officer, Director · decision-maker
Careers mailbox is known
Verified live · 3 hours ago 19 days ago Hiring now · 70%

Machine Learning Researcher - Agentic Science

≈ $139k – $304k per year (Estimated) • Remote (United States) • PhD • Full-Time
Python
AI/ML
JAX
Multimodal AI
AI Agents
PyTorch
In-Context Learning
Few-Shot Learning
Agentic Workflows
Machine Learning
+1
Apply
Report

Cerebras Systems

cerebras.ai
cerebras.ai • HQ: Sunnyvale, United States • Servers & Data Center Hardware • AI Compute & Inference • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2016 • NASDAQ: CBRS • Trust B • Hiring 119, 3 new
HQ: Sunnyvale, United States • Servers & Data Center Hardware • AI Compute & Inference • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2016 • NASDAQ: CBRS • Trust B • Hiring 119, 3 new
Cerebras Systems is an American computer hardware company founded in 2016 and headquartered in Sunnyvale, California that builds accelerators for artificial intelligence at wafer scale. Instead of assembling clusters from many small chips, it manufactures a single processor the size of an entire silicon wafer, the Wafer Scale Engine, which removes most of the communication overhead in large model training and inference. The company sells CS-series systems to research laboratories and enterprises, operates its own inference cloud known for very high token throughput, and has built large supercomputers with partners including the Gulf technology group G42.
Sean Lie · Co-Founder & Chief Technology Officer · decision-maker
Gary Lauterbach · Co-Founder & Chief Technology Officer · decision-maker
Andrew Feldman · Co-Founder & Chief Executive Officer · decision-maker
Jean-Philippe Fricker · Co-Founder & Chief System Architect · founder
Careers mailbox is known
Verified live · 12 min ago Visa sponsor 19 days ago Likely open · 61%

ML Algorithm Mapping and Performance Engineer, Core ML

≈ $198k – $433k per year (Estimated) • Hybrid (Sunnyvale, United States) • Bachelor's Degree • Full-Time
Python
C++
C++
PyTorch C++
AI/ML
AI Agents
Mixture of Experts
Cerebras
OpenAI
Edge AI
Speculative Decoding
Machine Learning
CUDA Toolkit
Quantization
JAX
PyTorch
LLM
CUDA
Triton
KV Cache
DevOps
HPC
+11
Apply
Verified live · 12 min ago Visa sponsor 19 days ago Likely open · 61%

ML Runtime and Kernel Engineer - Core ML

≈ $198k – $434k per year (Estimated) • Hybrid (Sunnyvale, United States, Toronto, Canada) • Bachelor's Degree • Full-Time
Python
C++
C++
PyTorch C++
AI/ML
JAX
AI Agents
PyTorch
LLM
Cerebras
OpenAI
Edge AI
Machine Learning
CUDA Toolkit
CUDA
Triton
KV Cache
DevOps
HPC
+8
Apply
Verified live · 12 min ago Visa sponsor 3 months ago Fading · 34%

Director/Sr. Manager, AI Inference Model Scaling

≈ $249k – $466k per year (Estimated) • Head+ • 5+ years exp • Hybrid (Sunnyvale, United States) • Bachelor's Degree • Full-Time • Senior
Python
C++
C++
LLVM
TensorFlow C++
PyTorch C++
AI/ML
AI Agents
LLM
Cerebras
OpenAI
Edge AI
MLIR
Apache TVM
XLA
JAX
ONNX
TensorFlow
PyTorch
Machine Learning
+10
Apply

Autodesk

autodesk.com
autodesk.com • HQ: San Francisco, United States • Simulation & Digital Twin Software • Construction Management Software • CAD, PLM & BIM Software • 5000+ employees • Est. 1982 • Trust A • Hiring 378, 43 new
HQ: San Francisco, United States • Simulation & Digital Twin Software • Construction Management Software • CAD, PLM & BIM Software • 5000+ employees • Est. 1982 • Trust A • Hiring 378, 43 new
Autodesk is an American software company founded in 1982 that makes the design and engineering tools used to create buildings, infrastructure, products, factories and visual effects. AutoCAD established it as the standard for computer-aided drafting, and the portfolio now spans Revit for building information modelling, Fusion for product design and manufacturing, Civil 3D for infrastructure, and Maya and 3ds Max for film, television and games. Headquartered in San Francisco, the company moved fully to subscription licensing in the last decade and is pushing customers toward cloud platforms that connect design, construction and production data across a project's lifecycle.
Carmel Galvin · Executive Officer · decision-maker
Pascal Di Fronzo · Executive Officer · decision-maker
Steven Blum · Executive Officer · decision-maker
Richard Herren · Executive Officer · decision-maker
Careers mailbox is known
Verified live · 9 hours ago Visa sponsor 20 days ago Likely open · 64%

AI Research Scientist

$122k – $179k per year • Equity • In office (Toronto, Canada) • Master's Degree • Full-Time
JAX
Multimodal AI
Computer Vision
Transformers
PyTorch
Tokenization
Post-training
Machine Learning
Reinforcement Learning
Embodied AI
Robotics
Reinforcement Learning
+3
Apply

Publicis Groupe

publicisgroupe.com
publicisgroupe.com • HQ: Paris, France • Creative Agencies • Digital Marketing • Advertising • 5000+ employees • Est. 1926 • Trust A • Hiring 1214, 148 new
HQ: Paris, France • Creative Agencies • Digital Marketing • Advertising • 5000+ employees • Est. 1926 • Trust A • Hiring 1214, 148 new
Publicis Groupe is a French multinational advertising and communications group headquartered in Paris that provides creative, media, data, commerce and digital transformation services to brands around the world. Founded in 1926 by Marcel Bleustein-Blanchet, it spans creative agency networks, Publicis Media, the data marketing company Epsilon, Publicis Sapient and Publicis Health, with around 100,000 employees in more than 100 countries. It hires account managers, media planners and analysts, SEO and paid media specialists, designers and content producers, and software, data and quality engineers for Publicis Sapient.
Céline Fronval · Directeur de la publication
Juridique Publicis · Directeur de la publication
Careers mailbox is known
Verified live · 2 hours ago 20 days ago Fading · 26%

Director Software Engineering

≈ $45k – $119k per year (Estimated) • Head+ • 15+ years exp • In office (Bengaluru, India) • Master's Degree • Architect
Vertex AI
JAX
Multimodal AI
Computer Vision
AI Agents
NLP
TensorFlow
PyTorch
Amazon SageMaker
Machine Learning
RLHF
RLAIF
DevOps
Azure
AWS
+6
Apply
Machines can read this selection · checked 21:52 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.