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
Software
Data
Security
DevOps & Cloud
Leadership
Product
Design
Fintech
Hardware & Chips
Defense
Aerospace
Biotech
Energy
Robotics
Automotive
MedTech
Science

NTT DATA Romania

nttdata.ro
nttdata.ro · HQ: Bucharest, Romania · 1001-5000 employees · Hiring 74, 12 new
HQ: Bucharest, Romania · 1001-5000 employees · Hiring 74, 12 new
NTT DATA Romania is the Romanian arm of the Japanese IT services group NTT DATA, delivering software development, cloud, data, SAP, Salesforce and cybersecurity projects for clients in Europe. It works from offices in Bucharest, Cluj-Napoca, Iasi, Brasov, Timisoara and Sibiu, on assignments such as European Commission customs digitalisation, telecom data platforms and automotive software, and its board also lists roles in Novi Sad and Rio de Janeiro. It hires Java, full-stack and frontend developers, data and platform engineers, DevOps and AUTOSAR engineers, SAP consultants, project managers and security specialists.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 day ago 31 days ago Fading · 39%

AI Engineer

Middle · 3+ years exp · In office (Iași, Romania) · Bachelor's Degree · Middle
SQL
Python
Databases
Weaviate
Chroma
pgvector
Pinecone
PostgreSQL
AI/ML
LangGraph
LlamaIndex
Fine-tuning
Embeddings
Scikit-learn
JAX
Function Calling
Chain-of-Thought
Google ADK
Tokenization
Human-in-the-Loop
Multi-Agent Systems
TensorFlow
LangChain
Prompt Engineering
PyTorch
RAG
LLM
AI Agents
Vertex AI
Gemini
Semantic Search
LLM Evaluation
DevOps
gRPC
Rest API
Git
GCP
Azure
CI/CD
AWS
OpenTelemetry
Datadog
Prometheus
Grafana
+33
Apply

ARONDITE

arondite.com
arondite.com · London · Oslo · Ottawa · Military · Defense Manufacturing · 11-50 employees · Trust A · Hiring 19, 1 new
London · Oslo · Ottawa · Military · Defense Manufacturing · 11-50 employees · Trust A · Hiring 19, 1 new
Arondite is the British defence tech company for an autonomous age. We build technology that delivers immediate, real-world impact on a global scale.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago Visa sponsor 31 days ago Likely open · 68%

AI Engineer (Agents)

≈ $75k – $201k per year (Estimated) · Equity · Hybrid (London, United Kingdom) · Relocation · Full-Time · Citizenship or vetting required
Rust
C++
Java
Python
C++
TensorFlow C++
AI/ML
LangGraph
JAX
Multimodal AI
Google ADK
Anthropic
TensorFlow
Model Context Protocol
PyTorch
Machine Learning
AI Agents
LangChain
Cybersecurity
Wiz
Design
Figma
+10
Apply
Verified live · 2 hours ago Visa sponsor 3 months ago Long shot · 22%

AI Engineer (Algorithms)

≈ $75k – $200k per year (Estimated) · Equity · Hybrid (London, United Kingdom) · Relocation · Full-Time · Citizenship or vetting required
Rust
C++
Java
Python
C++
TensorFlow C++
AI/ML
JAX
Multimodal AI
Synthetic Data
Anthropic
TensorFlow
Computer Vision
PyTorch
Machine Learning
Cybersecurity
Wiz
Robotics
Path Planning
Design
Figma
+8
Apply

Bentley Systems

bentley.com
bentley.com · HQ: Exton, United States · Simulation & Digital Twin Software · CAD, PLM & BIM Software · 5000+ employees · Est. 1984 · NASDAQ: BSY · Hiring 89, 8 new
HQ: Exton, United States · Simulation & Digital Twin Software · CAD, PLM & BIM Software · 5000+ employees · Est. 1984 · NASDAQ: BSY · Hiring 89, 8 new
Bentley Systems is an infrastructure engineering software company headquartered in Exton, Pennsylvania, whose products support the design, construction and operation of roads, rail, bridges, water networks, utilities and industrial assets. Founded in 1984 and listed on Nasdaq, it serves engineers, constructors and owner-operators worldwide and has major development and sales offices in Ireland, Lithuania, India, Canada and Australia. The company hires software engineers in C++ and other stacks, solution architects, pre-sales and application engineers, UX designers, product counsel and government relations specialists.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 min ago 31 days ago Fading · 25%

Applied AI Engineer - Engineering Simulation & Design Exploration

≈ $75k – $161k per year (Estimated) · Hybrid (Quebec, Canada) · Full-Time
Python
AI/ML
JAX
PyTorch
Machine Learning
DevOps
HPC
Robotics
Digital Twin
Apply

Salesforce

salesforce.com
salesforce.com · HQ: San Francisco, United States · AI Agents · Team Communication & Collaboration · Marketing Automation · Customer Support & Help Desk Software · BI & Dashboard Software · Marketing & Sales AI · CRM Software · 5000+ employees · Est. 1999 · Trust B · Hiring 1411, 188 new
HQ: San Francisco, United States · AI Agents · Team Communication & Collaboration · Marketing Automation · Customer Support & Help Desk Software · BI & Dashboard Software · Marketing & Sales AI · CRM Software · 5000+ employees · Est. 1999 · Trust B · Hiring 1411, 188 new
Salesforce is an American enterprise software company founded in 1999 that pioneered delivering business applications over the internet rather than as installed software. Its Customer 360 platform brings sales, service, marketing, commerce and analytics onto shared customer data, extended by the Slack collaboration suite, the Tableau analytics business and the MuleSoft integration layer. Headquartered in San Francisco, the company has moved its product strategy toward autonomous agents through Agentforce and the Data Cloud, and it remains the largest customer relationship management vendor in the world by revenue share.
Parker Harris · Co-Founder · decision-maker
Marc Benioff · Co-Founder, Chairman & Chief Executive Officer · decision-maker
Brent Hyder · Executive Officer · decision-maker
Srinivas Tallapragada · Executive Officer · decision-maker
No careers mailbox known
Top 25% payVerified live · 1 day ago 31 days ago Fading · 34%

Director of Product Management, Agentforce Voice Models

$197k – $314k per year · Head+ · Hybrid (San Francisco, Washington, United States) · PhD · Full-Time · Architect
Python
AI/ML
Fine-tuning
JAX
NLP
Speech Recognition
Agentforce
Text-to-Speech
Voice Agents
PyTorch
Machine Learning
AI Agents
Quantization
DevOps
WebRTC
GCP
Azure
AWS
Cybersecurity
HIPAA
GDPR
Analytics
A/B Testing
+11
Apply
Verified live · 1 day ago 1 month ago Long shot · 12%

AI Research Scientist, Singapore

≈ $58k – $162k per year (Estimated) · In office (Singapore) · Master's Degree · Full-Time
C++
Python
C++
TensorFlow C++
AI/ML
JAX
Agentforce
TensorFlow
PyTorch
Machine Learning
AI Agents
Multimodal AI
NLP
Time Series
Prompt Engineering
+5
Apply
Top 25% payVerified live · 1 day ago 2 months ago Long shot · 9%

Research Scientist - Salesforce AI Research

$117k – $314k per year · In office (Palo Alto, San Francisco, United States) · PhD · Full-Time
Python
AI/ML
DeepSpeed
RLHF
Reinforcement Learning
JAX
Multimodal AI
Speech Recognition
Time Series
DPO
Post-training
Pre-training
Agentforce
Text-to-Speech
Computer Use
Model Distillation
Computer Vision
Prompt Engineering
PyTorch
LLM
Machine Learning
AI Agents
+13
Apply

Optiver

optiver.com
optiver.com · HQ: Amsterdam, Netherlands · Hedge Funds & Alternative Investments · 1001-5000 employees · Est. 1986 · Trust A · Hiring 142, 5 new
HQ: Amsterdam, Netherlands · Hedge Funds & Alternative Investments · 1001-5000 employees · Est. 1986 · Trust A · Hiring 142, 5 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 · 2 hours ago Visa sponsor 1 month ago Hiring now · 89%

AI Lab - Research

$200k per year · In office (New York, United States)
Python
AI/ML
Reinforcement Learning
JAX
Mamba
Transformers
Self-Supervised Learning
Time Series
CUDA
Triton
Post-training
PyTorch
Machine Learning
+4
Apply
Verified live · 2 hours ago 2 months ago Likely open · 52%

2027 Shanghai Machine Learning Engineer Summer Internship

Intern · In office (Shanghai, China) · Internship · Intern
JAX
PyTorch
Machine Learning
MLFlow
Apply
Verified live · 2 hours ago Visa sponsor 8 months ago Long shot · 9%

Machine Learning Performance Engineer

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

xAI

x.ai
x.ai · HQ: Palo Alto, United States · AI Research Labs · Foundation Models · LLM & Generative AI · 1001-5000 employees · Est. 2023 · Trust A · Hiring 241, 7 new
HQ: Palo Alto, United States · AI Research Labs · Foundation Models · LLM & Generative AI · 1001-5000 employees · Est. 2023 · Trust A · Hiring 241, 7 new
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.
Elon Musk · Founder & Chief Executive Officer · decision-maker
Toby Pohlen · Co-Founder · founder
Jimmy Ba · Co-Founder · founder
Igor Babuschkin · Co-Founder · founder
No careers mailbox known
Top 25% payVerified live · 3 hours ago 1 month ago Likely open · 54%

Software Engineer, Evaluation Infrastructure

$180k – $440k per year · In office (Palo Alto, United States)
Rust
C++
Python
AI/ML
Grok
JAX
PyTorch
LLM
Apply
Top 25% payVerified live · 3 hours ago 3 months ago Long shot · 17%

ML Infrastructure Engineer

$440k per year · Junior · 2+ years exp · In office (Palo Alto, United States) · PhD · Junior
Rust
C++
Python
AI/ML
Machine Learning
JAX
CUDA
PyTorch
DevOps
Puppet
SLURM
Configuration Management
Ansible
Linux
+4
Apply
Top 25% payVerified live · 3 hours ago 3 months ago Long shot · 10%

Model Engineer, RL Training Framework

$180k – $440k per year · In office (Palo Alto, United States)
Rust
C++
Python
AI/ML
Grok
JAX
PyTorch
Reinforcement Learning
LLM
Apply

Xanadu

xanadu.ai
xanadu.ai · HQ: Toronto, Canada · Quantum Computing Hardware · Est. 2016 · Hiring 88, 5 new
HQ: Toronto, Canada · Quantum Computing Hardware · Est. 2016 · Hiring 88, 5 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 1 month ago Likely open · 59%

Quantum Scientist - Software

≈ $66k – $195k 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 · 33%

Quantum Scientist — Finance

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

Quantum Scientist - Compilation

≈ $62k – $185k per year (Estimated) · Hybrid (Toronto, Canada) · Bachelor's Degree · Full-Time
C++
Python
C++
LLVM
AI/ML
JAX
MLIR
Machine Learning
Quantum
PennyLane
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 4
HQ: Bengaluru, India · Document AI · Speech & Audio AI · EHR & Clinical Software · Medical AI · Est. 2025 · Likely startup · Hiring 4
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
1 month ago Fading · 26%

Lead AI Engineer - Speech & Clinical AI

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

Vantor

vantor.com
vantor.com · HQ: Westminster, United States · Space Services · Geospatial Analytics · Earth Observation · 201-500 employees · Est. 2025 · Trust B · Hiring 135, 15 new
HQ: Westminster, United States · Space Services · Geospatial Analytics · Earth Observation · 201-500 employees · Est. 2025 · Trust B · Hiring 135, 15 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 · 11 hours ago 1 month ago Likely open · 47%

Principal Applied AI Scientist

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

Applied AI Scientist

$128k – $170k per year · Senior · 8+ years exp · United States · Remote (United States) · Bachelor's Degree · Full-Time · Nationals only · Staff
Python
AI/ML
JAX
Multimodal AI
TensorFlow
Computer Vision
PyTorch
Machine Learning
Fine-tuning
AI Agents
DevOps
GCP
+2
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 670, 35 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 670, 35 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 1 month ago Likely open · 48%

Research Engineer / Research Scientist, RL Frontiers

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

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
JAX
Multimodal AI
Anthropic
Interpretability
TensorFlow
PyTorch
Claude
Machine Learning
Analytics
A/B Testing
+2
Apply
Open 190 daysTop 25% pay 6 months ago Long shot · 4%

Research Engineer, Machine Learning (RL Velocity)

$496k – $845k per year · Hybrid (London, United Kingdom) · Bachelor's Degree
JAX
Multimodal AI
Anthropic
Post-training
Pre-training
Interpretability
PyTorch
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 20, 1 new
HQ: San Francisco, United States · Machine Learning · AI for Science · Simulation & Digital Twin Software · 201-500 employees · Est. 2025 · Likely startup · Hiring 20, 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 1 month ago Fading · 31%

CAD Automation Researcher

≈ $164k – $365k per year (Estimated) · In office (San Francisco, United States) · PhD · Full-Time
Python
Databases
Databricks
AI/ML
Embeddings
Reinforcement Learning
JAX
Diffusion Models
PyTorch
SpaceTech
ANSYS
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 · 4 hours ago Visa sponsor 1 month ago Fading · 31%

(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
Pandas
NumPy
Self-Supervised Learning
Time Series
TensorFlow
PyTorch
Machine Learning
DevOps
GCP
AWS
+7
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
MATLAB
C++
Python
C++
TensorFlow C++
AI/ML
Machine Learning
JAX
TensorFlow
PyTorch
Apply
≈ $71k – $189k per year (Estimated) · Hybrid (Southampton, United Kingdom) · Master's Degree · Contractor
MATLAB
C++
Python
C++
TensorFlow C++
AI/ML
Machine Learning
JAX
TensorFlow
PyTorch
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 538, 49 new
HQ: London, United Kingdom · Data Providers & Datasets · Financial Data & Market Intelligence · Stock Exchanges & Market Infrastructure · 5000+ employees · Est. 2007 · Trust B · Hiring 538, 49 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 · 3 hours ago 1 month ago Fading · 32%

Enterprise Data Scientist

≈ $62k – $149k per year (Estimated) · Staff+ · 10+ years exp · In office (Mumbai, India) · Bachelor's Degree · Full-Time · Staff
SQL
Python
Python
Dask
Databases
Weaviate
Cassandra
Pinecone
DynamoDB
ElasticSearch
Apache Kafka
AI/ML
NLP
LLMOps
LLM Guardrails
RAG
Machine Learning
AI Agents
LangGraph
AutoGen
Qwen
Airflow
DeepSeek
LoRA
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
Pandas
NumPy
CrewAI
Gemini
BERT
NLTK
Ray
Hallucination
Hybrid Search
Sentiment Analysis
Hugging Face
GPT-5
Semantic Search
EU AI Act
Multi-Agent Systems
Model Distillation
TensorFlow
LangChain
Model Context Protocol
Spark
PyTorch
Claude
LLM
DevOps
GCP
Azure
AWS
Analytics
Plotly
Tableau
Power BI
+59
Apply

MARUM

marum.de
marum.de · HQ: Bremen, Germany · Research Institutes · Environmental Science · 501-1000 employees · Trust B · Hiring 10
HQ: Bremen, Germany · Research Institutes · Environmental Science · 501-1000 employees · Trust B · Hiring 10
MARUM is the Center for Marine Environmental Sciences, a research institution in marine environmental science under the supervision of the Bremen senator for education and science. It offers doctoral positions, internships and administrative roles, including PhD research posts in artificial intelligence and HR internships. The centre is based in Bremen, Germany.
No named decision-maker yet.
No careers mailbox known
Verified live · 11 hours ago 1 month ago Likely open · 57%

PhD Position (m/f/d): World Models for Software Architecture Synthesis (Constructor Fabric)

≈ $17k – $26k per year (Estimated) · In office (Bremen, Germany) · Bachelor's Degree · Full-Time · English
Python
Rust
Kotlin
C++
Java
Python
Asyncio
Dask
AI/ML
Weights & Biases
DGL
MLFlow
Ray
GNN
LLM Evaluation
World Models
PyTorch
LLM
Machine Learning
JAX
DevOps
Linux
OpenTelemetry
Prometheus
Terraform
Docker
Kubernetes
Robotics
Digital Twin
QA
Pytest
+18
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 1 month ago Likely open · 48%

Machine Learning Researcher - Agentic Science

≈ $137k – $305k per year (Estimated) · Remote (United States) · PhD · Full-Time
Python
AI/ML
JAX
Multimodal AI
In-Context Learning
Few-Shot Learning
Agentic Workflows
PyTorch
Machine Learning
AI Agents
+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 A · Hiring 113, 1 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 A · Hiring 113, 1 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 · 4 hours ago Visa sponsor 1 month ago Hiring now · 85%

ML Algorithm Mapping and Performance Engineer, Core ML

≈ $201k – $448k per year (Estimated) · Hybrid (Sunnyvale, United States) · Bachelor's Degree · Full-Time
C++
Python
AI/ML
Mixture of Experts
Cerebras
Speculative Decoding
OpenAI
Machine Learning
Quantization
JAX
CUDA
Triton
KV Cache
PyTorch
LLM
DevOps
HPC
+7
Apply
Verified live · 4 hours ago Visa sponsor 1 month ago Hiring now · 85%

ML Runtime and Kernel Engineer - Core ML

≈ $203k – $452k per year (Estimated) · Hybrid (Sunnyvale, United States, Toronto, Canada) · Bachelor's Degree · Full-Time
C++
Python
AI/ML
JAX
Cerebras
OpenAI
PyTorch
LLM
Machine Learning
CUDA
Triton
KV Cache
DevOps
HPC
+4
Apply
Verified live · 4 hours ago Visa sponsor 3 months ago Fading · 38%

Director/Sr. Manager, AI Inference Model Scaling

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

Qualcomm

qualcomm.com
qualcomm.com · HQ: San Diego, United States · Autonomous Driving · Car Electronics · RF & Wireless Hardware · Processors, MCUs & AI Chips · 5000+ employees · Est. 1985 · Trust C · Hiring 1817, 46 new
HQ: San Diego, United States · Autonomous Driving · Car Electronics · RF & Wireless Hardware · Processors, MCUs & AI Chips · 5000+ employees · Est. 1985 · Trust C · Hiring 1817, 46 new
Qualcomm is an American semiconductor and wireless technology company founded in 1985 and headquartered in San Diego, California. It holds foundational patents in CDMA and the 3G, 4G and 5G cellular standards, licensing them to nearly every handset maker in the world, while its chip division designs the Snapdragon platforms that power a large share of Android smartphones, personal computers, wearables and connected cars. The company has been diversifying away from smartphone dependence into automotive digital cockpit and driver assistance systems, industrial internet of things, and on-device AI processing through its Hexagon and Oryon architectures.
Akash Palkhiwala · Chief Financial Officer & Chief Operating Officer · decision-maker
Cristiano Amon · President & Chief Executive Officer · decision-maker
Andrew Viterbi · Co-Founder · founder
Irwin Jacobs · Co-Founder & former Chairman · founder
No careers mailbox known
Verified live · 3 hours ago 1 month ago Hiring now · 80%

Performance & Modeling Engineer

≈ $38k – $85k per year (Estimated) · Senior · 5+ years exp · In office (Shanghai, China) · Bachelor's Degree · Senior · English: B2
C++
Python
Java
C++
TensorFlow C++
AI/ML
vLLM
JAX
Multimodal AI
SGLang
ONNX
TensorRT-LLM
TGI
Transformers
Mixture of Experts
TensorFlow
PyTorch
LLM
llama.cpp
Quantization
TensorRT
Ollama
MLC LLM
Speculative Decoding
KV Cache
DevOps
Linux
Docker
Kubernetes
QA
Rest-Assured
+19
Apply
Verified live · 3 hours ago 1 month ago Hiring now · 80%

Software Integration & Test Engineer

≈ $34k – $74k per year (Estimated) · Senior · 5+ years exp · In office (Shanghai, China) · Bachelor's Degree · Senior · English: B2
Python
C++
Java
C++
TensorFlow C++
AI/ML
JAX
Multimodal AI
ONNX
Transformers
Mixture of Experts
TensorFlow
PyTorch
Machine Learning
vLLM
Quantization
SGLang
TensorRT
TensorRT-LLM
TGI
TPU
Speculative Decoding
KV Cache
LLM
DevOps
Linux
Docker
Kubernetes
CI/CD
QA
Rest-Assured
+19
Apply

Publicis Groupe

publicisgroupe.com
publicisgroupe.com · HQ: Paris, France · Creative Agencies · Digital Marketing · Advertising · 5000+ employees · Est. 1926 · Trust A · Hiring 1279, 211 new
HQ: Paris, France · Creative Agencies · Digital Marketing · Advertising · 5000+ employees · Est. 1926 · Trust A · Hiring 1279, 211 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 · 1 hour ago 1 month ago Fading · 26%

Director Software Engineering

≈ $43k – $88k per year (Estimated) · Head+ · 15+ years exp · In office (Bengaluru, India) · Master's Degree · Architect
Vertex AI
JAX
Multimodal AI
NLP
Amazon SageMaker
TensorFlow
Computer Vision
PyTorch
Machine Learning
AI Agents
RLHF
RLAIF
DevOps
Azure
AWS
+6
Apply
allenai.org · HQ: Seattle, United States · Research Institutes · Foundation Models · Natural Language Processing · AI Research Labs · Est. 2014 · Trust A · Hiring 16, 1 new
HQ: Seattle, United States · Research Institutes · Foundation Models · Natural Language Processing · AI Research Labs · Est. 2014 · Trust A · Hiring 16, 1 new
Headquartered in Seattle, Washington, the Allen Institute for AI (AI2) is a non-profit research institute founded by Microsoft co-founder Paul Allen to conduct high-impact artificial intelligence research and engineering in the public interest. The organization focuses on advancing open science, developing large-scale natural language processing models, and building foundational AI systems for scientific discovery and environmental conservation. Additionally, it freely releases its datasets, open-source models, and research papers, alongside creating tools like Semantic Scholar to accelerate global scientific research.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 1 hour ago 1 month ago Hiring now · 89%

Research Engineer, Climate Modeling

$147k – $220k per year · Junior · 2+ years exp · In office (Seattle, United States) · Bachelor's Degree · Junior
Python
AI/ML
JAX
TPU
Machine Learning
DevOps
GitHub
GCP
Apply
Top 25% payVerified live · 1 hour ago 1 month ago Hiring now · 90%

Research Scientist, Climate Modeling

$167k – $261k per year · In office (Seattle, United States) · PhD
Python
AI/ML
JAX
TPU
Machine Learning
Apply
Top 25% payVerified live · 1 hour ago 4 months ago Fading · 32%

Senior Research Engineer, Olmo + Molmo

$174k – $261k per year · Senior · 4+ years exp · In office (Seattle, United States) · Bachelor's Degree · Senior
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
Function Calling
LLaVA
Agentic Workflows
TensorFlow
PyTorch
LLM
AI Agents
OpenAI
DevOps
Docker
GCP
AWS
+7
Apply
Machines can read this selection · checked 15:22 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.