JAX Jobs - Remote & On-site

JAX jobs at vetted startups and product companies focused on high-performance ML and research, updated daily. These roles often pay competitively and support remote or hybrid work arrangements. Browse and apply today.

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

Blueorigin

blueorigin.com
Blue Origin is an American aerospace manufacturer, defense contractor, and spaceflight services company founded in 2000 by Jeff Bezos. Headquartered in Kent, Washington, the company designs and operates reusable rockets, heavy-lift engines, and deep-space logistics infrastructure to enable sustainable space exploration and commercial operations.
blueorigin.com • Los Angeles • Washington • Seattle • Huntsville • Arlington • Space Services • Space & Aerospace • Launch Services • 1001-5000 employees
Los Angeles • Washington • Seattle • Huntsville • Arlington • Space Services • Space & Aerospace • Launch Services • 1001-5000 employees
Open 92 daysTop 25% pay 3 months ago

Chief Engineer - ML, Physical AI, & Autonomy

$254k – $355k per year • Equity • Executive • 10+ years expIn office (Seattle, United States) • Master's Degree • Full-Time • Architect
C++
Python
MATLAB
C++
PyTorch C++
TensorFlow C++
MATLAB
Simulink
AI/ML
Computer Vision
JAX
PyTorch
TensorFlow
AI Agents
Robotics
Digital Twin
Sensor Fusion
Gazebo
Isaac Sim
MuJoCo
ROS
ROS2
Apply
Report

AppLike Group

applike-group.com
AppLike Group is a Hamburg company founded in 2014 that builds mobile advertising and app discovery products. Its portfolio includes user acquisition platforms, rewarded engagement apps and monetisation technology for publishers. The company grew out of the Bertelsmann backed venture ecosystem in Hamburg.
applike-group.com • HQ: Hamburg, Germany • Professional Services • 201-500 employees • Est. 2014
HQ: Hamburg, Germany • Professional Services • 201-500 employees • Est. 2014
Open 94 daysVerified live · 1 day ago 4 months ago

Senior Machine Learning Engineer (f/m/d)

$71k – $179k per year (Estimated) • Senior • 5+ years expRemote/Hybrid (Hamburg, Germany) • Relocation • Full-Time • Senior
Go
Databases
Apache Kafka
MySQL
Trino
AI/ML
Flink
JAX
PyTorch
Ray
TensorFlow
Recommender Systems
DevOps
AWS
Amazon S3
Analytics
A/B Testing
Apply
Report

Hugging Face

huggingface.co
Hugging Face is an American company, founded in 2016 and headquartered in New York, that runs the largest open platform for sharing machine learning models, datasets and demonstration applications. Its Hub hosts more than a million public models and is often described as the GitHub of machine learning, while its open-source libraries such as Transformers, Diffusers and Datasets are standard tools in research and production pipelines. The company earns revenue from paid Hub features, managed inference endpoints and enterprise deployments, and works with cloud and chip vendors to make open models easy to run anywhere.
huggingface.co • HQ: New York, United States • Software • MLOps • Machine Learning • LLM & Generative AI • Artificial Intelligence • 201-500 employees • Est. 2016
HQ: New York, United States • Software • MLOps • Machine Learning • LLM & Generative AI • Artificial Intelligence • 201-500 employees • Est. 2016
Open 95 daysVerified live · 2 days ago 4 months ago

Open-Source Machine Learning Engineer - US Remote

$140k – $307k per year (Estimated)New York • Remote (United States) • Full-Time
Python
AI/ML
Accelerate
Gradio
JAX
PyTorch
TensorFlow
Transformers
vLLM
Hugging Face
Fine-tuning
DevOps
GitHub
Management
Slack
Apply
Open 95 daysVerified live · 2 days ago 4 months ago

Open-Source Machine Learning Engineer - EMEA Remote

$69k – $165k per year (Estimated)Paris • Remote (EMEA) • Full-Time
Python
AI/ML
Accelerate
Gradio
JAX
PyTorch
TensorFlow
Transformers
vLLM
Hugging Face
Fine-tuning
DevOps
GitHub
Management
Slack
Apply
Report

Machinify

machinify.com
Machinify is an artificial intelligence software company headquartered in Palo Alto, California, and founded in 2016. The company provides an AI-powered operating system designed to automate healthcare payment integrity processes, including claims auditing, subrogation, and pharmacy payment accuracy. It serves major health plans across the United States, managing data for over 270 million members to identify cost savings and reduce administrative friction for payers and providers.
machinify.com • HQ: Palo Alto, United States • Business Process Automation (BPA) • Software • Insurance • Financial Services • Machine Learning • Health Care • Digital Health • Artificial Intelligence • Est. 2016
HQ: Palo Alto, United States • Business Process Automation (BPA) • Software • Insurance • Financial Services • Machine Learning • Health Care • Digital Health • Artificial Intelligence • Est. 2016
Top 25% payVerified live · 2 hours ago 7 months ago

Staff Data Scientist | NLP

$180k – $230k per year • Staff+ • 3+ years exp • Remote (United States) • Staff
Python
SQL
Databases
pgvector
Pinecone
Weaviate
PostgreSQL
AI/ML
Fine-tuning
JAX
LLM
LoRA
NLP
PyTorch
QLoRA
RAG
Semantic Search
Sentence-Transformers
spaCy
Transformers
PEFT
AI Agents
Hugging Face
Human-in-the-Loop
LLM Evaluation
Semantic Search
Multimodal AI
DevOps
AWS
Azure
GCP
Vector
Analytics
A/B Testing
Apply
Report

Workday

workday.com
Workday is a leading provider of cloud-based enterprise software specializing in human capital management, financial management, and spend management. Founded in 2005, the company offers a unified platform that integrates artificial intelligence to help organizations streamline HR processes and optimize financial planning. Global corporations, educational institutions, and government agencies widely rely on its scalable solutions to manage workforce operations and operational data in real time.
workday.com • Pleasanton • Dublin • Pune • Amsterdam • Tokyo • AI Agents • Professional Services • Artificial Intelligence • 1001-5000 employees
Pleasanton • Dublin • Pune • Amsterdam • Tokyo • AI Agents • Professional Services • Artificial Intelligence • 1001-5000 employees
Open 108 daysTop 25% pay 4 months ago

Principal AI Researcher

$228k – $342k per year • Equity • Lead • 8+ years expIn office (Pleasanton, Atlanta, Seattle, United States) • Master's Degree • Full-Time • Principal
JAX
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
TensorFlow
DPO
Post-training
Pre-training
AI Agents
Apply
Report

Wayve

wayve.ai
Wayve is a British autonomous driving technology company that develops end-to-end "Embodied AI" foundation models for self-driving vehicles and advanced driver assistance systems (ADAS). Founded in 2017 by University of Cambridge researchers, Wayve pioneered what it calls the "AV2.0" approach to autonomous vehicles. Unlike traditional self-driving stacks that rely on rigid hand-coded rules, high-definition (HD) 3D maps, and specialized sensors, Wayve uses a unified deep learning model.
wayve.ai • HQ: London, United Kingdom • Transportation & Logistics • Ride Hailing • Artificial Intelligence • Autonomous Driving • 201-500 employees
HQ: London, United Kingdom • Transportation & Logistics • Ride Hailing • Artificial Intelligence • Autonomous Driving • 201-500 employees
Top 25% payVerified live · 4 hours ago 4 months ago

Research Scientist, Wayve Labs

$230k – $380k per year • Equity • Middle • 3+ years expIn office (Sunnyvale, United States) • Relocation • Visa sponsorship • Master's Degree • Middle
Python
AI/ML
Computer Vision
Ignite
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
Synthetic Data
Mixture of Experts
DeepSpeed
JAX
FSDP
Robotics
Reinforcement Learning
Sim-to-Real
Apply
Verified live · 4 hours ago 4 months ago

Research Scientist, Wayve Labs

$143k – $303k per year (Estimated) • Middle • 3+ years expRemote/Hybrid (Vancouver, Canada) • Relocation • Visa sponsorship • Master's Degree • Middle
Python
AI/ML
Computer Vision
Ignite
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
Synthetic Data
Mixture of Experts
DeepSpeed
JAX
FSDP
Robotics
Reinforcement Learning
Sim-to-Real
Apply
Verified live · 4 hours ago 4 months ago

Research Scientist, Wayve Labs

$111k – $309k per year (Estimated) • Middle • 3+ years expRemote/Hybrid (London, United Kingdom) • Relocation • Visa sponsorship • Master's Degree • Middle
Python
AI/ML
Computer Vision
Ignite
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
Synthetic Data
Mixture of Experts
DeepSpeed
JAX
FSDP
Robotics
Reinforcement Learning
Sim-to-Real
Apply
Report

Binance

binance.com
Binance is the world's largest cryptocurrency exchange by trading volume, founded in 2017 by Changpeng Zhao and Yi He. The platform offers spot, margin and derivatives trading across hundreds of digital assets, alongside staking, savings products, payments, an institutional custody arm and a self-custodial Web3 wallet. The group also created BNB Chain, one of the most used smart contract networks, and now operates under a licensed regional structure after a 2023 settlement with United States authorities that installed new leadership and compliance oversight.
binance.com • HQ: Abu Dhabi, United Arab Emirates • Mining & Staking • Crypto Payments • Web3 & dApps • Blockchain & Crypto • Cryptocurrencies • Crypto Exchanges • 1001-5000 employees • Est. 2017
HQ: Abu Dhabi, United Arab Emirates • Mining & Staking • Crypto Payments • Web3 & dApps • Blockchain & Crypto • Cryptocurrencies • Crypto Exchanges • 1001-5000 employees • Est. 2017
Verified live · 4 hours ago 4 months ago

Quantitative Researcher (Fresh STEM PhD graduates are welcome)

Hong Kong • Remote (Taiwan, Hong Kong) • PhD
Python
AI/ML
AI Agents
JAX
LLM
NumPy
PyTorch
Time Series Forecasting
Apply
Open 494 daysVerified live · 4 hours ago 1 year ago

Senior Large Language Model Algorithm Engineer/Expert

$98k – $247k per year (Estimated) • Senior • Remote/Hybrid • Master's Degree • Full-Time • Senior
Python
AI/ML
LLM
Reinforcement Learning
JAX
PyTorch
TensorFlow
AI Agents
DevOps
Vector
Web3
DeFi
Analytics
ETL/ELT
Apply
Report

Harmonic

harmonic.fm
Built for Artists. Curated by Fans. Create your own record store now.
harmonic.fm • Palo Alto • London • Publishing • Commerce • Marketplaces
Palo Alto • London • Publishing • Commerce • Marketplaces
Open 116 daysTop 25% pay 4 months ago

Research Engineer, Training & Inference

$200k – $450k per year • Senior • 5+ years expIn office (Palo Alto, United States) • Bachelor's Degree • Full-Time • Senior
Python
C++
C++
PyTorch C++
TensorFlow C++
AI/ML
CUDA
CUDA Toolkit
JAX
LLM
PyTorch
Reinforcement Learning
TensorFlow
NCCL
Triton
FSDP
Apply
Report

Abaka

abaka.ai
Abaka AI offers data collection, data cleaning, data annotation, and high-quality datasets for world-class Automobile AI, Generative AI, and Embodied AI industry leaders.
abaka.ai • Mountain View • Data Collection • LLM & Generative AI • Artificial Intelligence
Mountain View • Data Collection • LLM & Generative AI • Artificial Intelligence
Open 300 daysTop 25% pay 10 months ago

Machine Learning Engineer

$175k – $275k per year • Middle • 3+ years expIn office (Mountain View, United States) • Master's Degree • Middle
Python
AI/ML
Diffusion Models
Fine-tuning
JAX
MLFlow
Multimodal AI
PyTorch
Ray
TensorFlow
DevOps
AWS
Azure
GCP
Kubernetes
Apply
Report

Basecamp Research

basecamp-research.com
Basecamp Research is a London biotechnology company founded in 2019 that builds the biodiversity datasets used to train protein and enzyme design models. It collects samples through partnerships with countries and protected sites under benefit-sharing agreements, then sequences and structures them into a proprietary knowledge graph. The data underpins its own generative protein models and is licensed to pharmaceutical and industrial biotechnology partners.
basecamp-research.com • HQ: London, United Kingdom • Biotechnology • Est. 2019
HQ: London, United Kingdom • Biotechnology • Est. 2019
Open 123 daysVerified live · 1 day ago 5 months ago

Senior Research Engineer, ML Systems

$98k – $204k per year (Estimated) • Senior • In office (London, United Kingdom) • PhD • Full-Time • Senior
CUDA
CUDA Toolkit
JAX
MLX ML
PyTorch
Dagster
DevOps
Platform Engineering
Kubernetes
Management
Basecamp
Apply
Report

Spotify

spotify.com
Spotify is an audio streaming platform that delivers an extensive catalog of music, podcasts, and audiobooks, catering to hundreds of millions of users globally. It empowers artists by providing them with the means to monetize their creativity while offering fans unparalleled access to a diverse array of content, all driven by a passion for sound.
spotify.com • HQ: Stockholm, Sweden • Podcasting • Audio Production • Music Industry • Streaming & OTT Platforms • 1001-5000 employees
HQ: Stockholm, Sweden • Podcasting • Audio Production • Music Industry • Streaming & OTT Platforms • 1001-5000 employees
Verified live · 1 day ago 5 months ago

Senior Staff Machine Learning Engineer, Content Platform

$132k – $290k per year (Estimated) • Staff+ • Remote/Hybrid (London, United Kingdom) • Full-Time • Staff
JAX
Multimodal AI
PyTorch
TensorFlow
Apply
Report

Lovable

lovable.dev
Lovable is an AI-powered software development platform headquartered in Stockholm, Sweden. Founded in 2023 by Anton Osika and Fabian Hedin, the company operates a conversational "vibe coding" environment that enables both non-technical creators and software engineers to build, iterate, and deploy full-stack web applications using natural language prompts. Lovable automates front-end UI generation, back-end code creation, database connections, and GitHub workflow syncing, allowing users to move rapidly from product idea to production-ready application.
lovable.dev • HQ: Stockholm, Sweden • Cybersecurity • Design & Creative • 501-1000 employees • Est. 2023
HQ: Stockholm, Sweden • Cybersecurity • Design & Creative • 501-1000 employees • Est. 2023
Open 125 daysVerified live · 3 hours ago 5 months ago

AI Research Engineer, Post-Training

In office (Stockholm, Sweden, London, United Kingdom) • Full-Time
Fine-tuning
JAX
PyTorch
Reinforcement Learning
Lovable
Post-training
SFT
AI Agents
Marketing
Zendesk
Apply
Report

Constellation

constellation.space
ConstellationOS is the ML-native operations platform for satellite fleets. Unified telemetry, link forecasts, and policy-bound orchestration. Start with a 30-day shadow pilot.
constellation.space • Seattle • Software • Space & Aerospace • Ground Systems
Seattle • Software • Space & Aerospace • Ground Systems
$72k – $84k per year • Intern • In office (Seattle, United States) • Master's Degree • Internship • Junior
Python
AI/ML
Computer Vision
JAX
PyTorch
Reinforcement Learning
TensorFlow
Apply
Report

The Voleon Group

voleon.com
The Voleon Group is a premier quantitative investment management firm headquartered in Berkeley, California, that specializes in machine learning-driven trading strategies. Founded in 2007 by scientists Michael Kharitonov and Jon McAuliffe, the hedge fund combines advanced data science, statistical modeling, and computational AI to forecast financial markets and execute systematic algorithmic trades.
voleon.com • HQ: Berkeley, United States • Machine Learning • Financial Services • Asset Management
HQ: Berkeley, United States • Machine Learning • Financial Services • Asset Management
Top 25% payVerified live · 1 day ago 5 months ago

Senior Cluster Site Reliability Engineer

$205k – $235k per year • Senior • 5+ years expRemote/Hybrid (Berkeley, United States) • Bachelor's Degree • Full-Time • Senior
Python
Ruby
Python
Dask
AI/ML
Kubeflow
MLFlow
DeepSpeed
JAX
Modin
PyTorch
Ray
Spark
TensorFlow
InfiniBand
DevOps
Ansible
AWS
Configuration Management
GCP
Grafana
Loki
OpenTelemetry
Prometheus
SLURM
Terraform
Amazon S3
HPC
Docker
Kubernetes
Podman
IAM
Cybersecurity
Zero Trust
Apply
Report

Zencore

zencore.dev
Zencore is a leading cloud consulting and services firm. Above all else we value experience, hard work and the ability to keep things in perspective and have fun as we ensure our customers are delighted with the work we do.
zencore.dev • London • Mexico City • New York • Copenhagen • Berlin • Blockchain & Crypto • Data & Analytics • Information Technology • Est. 2013
London • Mexico City • New York • Copenhagen • Berlin • Blockchain & Crypto • Data & Analytics • Information Technology • Est. 2013
Open 137 days 5 months ago

Principal Architect, AI/ML

$139k – $249k per year (Estimated) • Lead • Remote (United States) • Master's Degree • Full-Time • Architect
AI Agents
Claude
Fine-tuning
Gemini
Google ADK
JAX
LangChain
Langfuse
LangGraph
LangSmith
Llama
LLM
LoRA
Mistral
PyTorch
Quantization
TensorRT
TensorRT-LLM
Vertex AI
vLLM
PEFT
LLM Guardrails
TPU
DevOps
AWS
Azure
GCP
Google GKE
Kubernetes
Cybersecurity
GDPR
Apply
Open 202 days 7 months ago

Principal Architect, AI/ML

$80k – $190k per year (Estimated) • Lead • Remote (United Kingdom) • Master's Degree • Full-Time • Architect
AI Agents
Claude
Fine-tuning
Gemini
Google ADK
JAX
LangChain
Langfuse
LangGraph
LangSmith
Llama
LLM
LoRA
Mistral
PyTorch
Quantization
TensorRT
TensorRT-LLM
Vertex AI
vLLM
PEFT
LLM Guardrails
TPU
DevOps
AWS
Azure
GCP
Google GKE
Kubernetes
Cybersecurity
GDPR
Apply
Report

Micron Technology

micron.com
Micron Technology is an American semiconductor company founded in Boise, Idaho in 1978 and one of only a handful of manufacturers of both DRAM and NAND flash memory. It designs and fabricates memory and storage products used in data centres, personal computers, smartphones, cars and industrial equipment, and sells them under the Micron and consumer-facing Crucial brands. High-bandwidth memory for AI accelerators has become the fastest-growing part of the business, and the company operates fabrication and assembly sites across the United States and Asia.
micron.com • HQ: Boise, United States • Data Centers • Semiconductor Manufacturing • Consumer Electronics • Hardware • Storage • Digital Storage • Semiconductors • 5000+ employees • Est. 1978
HQ: Boise, United States • Data Centers • Semiconductor Manufacturing • Consumer Electronics • Hardware • Storage • Digital Storage • Semiconductors • 5000+ employees • Est. 1978
Open 138 daysVerified live · 1 hour ago 5 months ago

Senior/Staff Engineer, Technology - AMAIS

$31k – $75k per year (Estimated) • Staff+ • 3+ years expIn office (Hyderabad, India) • Full-Time • Staff
Python
SQL
Python
pySpark
AI/ML
AutoGen
CrewAI
Google ADK
JAX
LangChain
LangGraph
MLFlow
PyTorch
Ray
Reinforcement Learning
RLHF
Scikit-learn
TensorFlow
Spark
AI Agents
Hugging Face
OpenAI
DevOps
GCP
Analytics
Tableau
Apply
Report

webAI

webai.com
webAI is an artificial intelligence platform company headquartered in Austin, Texas, and founded in 2019 by David Stout, Tyler Mauer, and Ethan Baird. The company builds software that trains and runs large models on hardware the customer already owns, including its Navigator model building tool, webFrame for spreading models across local clusters, and a device native assistant. It targets organizations in aviation, healthcare, manufacturing, financial services, and the public sector that need AI without sending data to a public cloud.
webai.com • HQ: Austin, United States • IT Infrastructure • Information Technology • AI Infrastructure • Hardware • LLM & Generative AI • Edge AI • Artificial Intelligence • Est. 2019
HQ: Austin, United States • IT Infrastructure • Information Technology • AI Infrastructure • Hardware • LLM & Generative AI • Edge AI • Artificial Intelligence • Est. 2019
Verified live · 10 hours ago 5 months ago

Director of Applied AI

$180k – $343k per year (Estimated) • Equity • Executive • 8+ years expIn office (Austin, United States) • Bachelor's Degree • Full-Time • Architect
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
PyTorch
TensorFlow
Edge AI
DevOps
Platform Engineering
Apply
Verified live · 10 hours ago 5 months ago

Senior Forward Deployed AI Engineer

$150k – $291k per year (Estimated) • Equity • Senior • In office (Austin, Washington, United States) • Master's Degree • Full-Time • Senior
JAX
ONNX
PyTorch
TensorFlow
AI Agents
Function Calling
Apply
Open 244 daysVerified live · 10 hours ago 9 months ago

AI Research Scientist

$144k – $270k per year (Estimated) • Equity • Middle • 4+ years expIn office (Austin, United States) • Bachelor's Degree • Full-Time • Middle
Python
AI/ML
Diffusion Models
Fine-tuning
JAX
Knowledge Distillation
Multimodal AI
PyTorch
Quantization
TensorFlow
Edge AI
Federated Learning
Reinforcement Learning
DevOps
Platform Engineering
Apply
Report

Yuno

yuno.ai
Yuno is an educational technology company based in San Francisco, California, and was founded in 2023. The organization develops an artificial intelligence-driven tutoring platform that provides personalized learning experiences and study assistance for students and professionals. It operates as a digital platform focusing on the integration of large language models to enhance academic performance and knowledge retention.
yuno.ai • HQ: San Francisco, United States • Higher Education • Secondary Education • Conversational AI • LLM & Generative AI • Education • Artificial Intelligence • EdTech Platforms • 201-500 employees • Est. 2023
HQ: San Francisco, United States • Higher Education • Secondary Education • Conversational AI • LLM & Generative AI • Education • Artificial Intelligence • EdTech Platforms • 201-500 employees • Est. 2023
Open 138 daysVerified live · 1 day ago 5 months ago

Senior AI Engineer

$28k – $72k per year (Estimated) • Senior • 5+ years expIn office (Hyderabad, Pakistan) • Senior
Python
AI/ML
AI Agents
Claude
Fine-tuning
Gemini
JAX
LangChain
Langfuse
LangGraph
LangSmith
LLM
LoRA
PyTorch
RAG
TensorFlow
PEFT
Context Engineering
SFT
LlamaIndex
Hugging Face
DevOps
AWS
Azure
GCP
Vector
CI/CD
Apply
Report

H Company

hcompany.ai
H Company is an artificial intelligence research company headquartered in Paris, France, and founded in 2023 by Charles Kantor with a founding team drawn largely from Google DeepMind. The company develops agentic AI systems, including Runner H, an agent that carries out multi-step tasks in a browser on the user's behalf. It raised one of the largest seed rounds in European history and positions itself as a French alternative in the agent tooling market.
hcompany.ai • HQ: Paris, France • Business Process Automation (BPA) • Software • Multimodal AI • LLM & Generative AI • AI Agents • Artificial Intelligence • 201-500 employees • Est. 2023
HQ: Paris, France • Business Process Automation (BPA) • Software • Multimodal AI • LLM & Generative AI • AI Agents • Artificial Intelligence • 201-500 employees • Est. 2023
Open 139 daysVerified live · 1 day ago 5 months ago

Research Engineer, Model Inference & Serving - Paris

$77k – $183k per year (Estimated)Remote/Hybrid (Paris, France) • Full-Time
C++
Python
Rust
C++
PyTorch C++
AI/ML
AI Agents
JAX
Multimodal AI
PyTorch
SGLang
vLLM
Edge AI
CUDA Toolkit
llama.cpp
LLM
MLX ML
ONNX
Quantization
TensorRT
TensorRT-LLM
CUDA
Triton
DevOps
Kubernetes
Apply
Open 139 daysVerified live · 1 day ago 5 months ago

Research Engineer / Scientist, Post-training & Reinforcement Learning - London

$104k – $219k per year (Estimated)Remote/Hybrid (London, United Kingdom) • Master's Degree • Full-Time
Python
Rust
AI/ML
AI Agents
Fine-tuning
JAX
Knowledge Distillation
Multimodal AI
NLP
PyTorch
Reinforcement Learning
RLHF
Synthetic Data
TensorFlow
DPO
Post-training
SFT
Function Calling
Mixture of Experts
Apply
Open 143 daysVerified live · 1 day ago 5 months ago

Research Engineer, Model Inference & Serving - London

$104k – $219k per year (Estimated)Remote/Hybrid (London, United Kingdom) • Full-Time
C++
Python
Rust
C++
PyTorch C++
AI/ML
AI Agents
JAX
Multimodal AI
PyTorch
SGLang
vLLM
Edge AI
CUDA Toolkit
llama.cpp
LLM
MLX ML
ONNX
Quantization
TensorRT
TensorRT-LLM
CUDA
Triton
DevOps
Kubernetes
Apply
Report

Perplexity AI

perplexity.ai
Perplexity AI is an American company that builds an answer engine combining live web search with large language models to return sourced, conversational responses instead of a list of links. Its products span a consumer assistant on web and mobile, the Comet browser, enterprise search over internal documents and the Sonar developer API that exposes the same grounded retrieval stack. Founded in 2022 in San Francisco by former researchers and engineers from OpenAI, Meta and Databricks, the company is backed by NVIDIA, IVP, New Enterprise Associates and SoftBank.
perplexity.ai • HQ: San Francisco, United States • AI Agents • Internet Services • Artificial Intelligence • Search Engines • LLM & Generative AI • 1001-5000 employees • Est. 2022
HQ: San Francisco, United States • AI Agents • Internet Services • Artificial Intelligence • Search Engines • LLM & Generative AI • 1001-5000 employees • Est. 2022
Open 140 daysTop 25% pay 5 months ago

Member of Technical Staff (AI Inference Engineer)

$220k – $485k per year • Staff+ • 3+ years expIn office (San Francisco, New York, Palo Alto, United States) • Full-Time • Staff
Python
Rust
AI/ML
CUDA Toolkit
Multimodal AI
Perplexity
CUDA
Triton
JAX
LLM
PyTorch
Quantization
TensorFlow
InfiniBand
NCCL
NVLink
Frontend
Sass
DevOps
API Gateway
Kubernetes
Apply
Open 140 daysVerified live · 28 min ago 5 months ago

Member of Technical Staff (AI Inference Engineer)

$161k – $354k per year (Estimated) • Staff+ • 3+ years expIn office (London, United Kingdom) • Full-Time • Staff
Python
Rust
AI/ML
CUDA Toolkit
Multimodal AI
Perplexity
CUDA
Triton
JAX
LLM
PyTorch
Quantization
TensorFlow
InfiniBand
NCCL
NVLink
Frontend
Sass
DevOps
API Gateway
Kubernetes
Apply
Open 145 daysVerified live · 28 min ago 5 months ago

Internship - Search Machine Learning Engineer

Intern • In office (London, United Kingdom) • Internship • Intern
Python
Rust
AI/ML
JAX
LLM
Perplexity
PyTorch
RAG
TensorFlow
Recommender Systems
NLP
Analytics
A/B Testing
Apply
Report
JAX Jobs - Remote & On-site
Frequently asked questions
JAX: How many jobs are available now?
There are 491 JAX jobs listed on Alion right now.
JAX: What is the average salary for these roles?
The average salary for JAX roles on Alion is $285,074 USD.
JAX: Are remote, relocation, or visa sponsorship options available?
Many JAX listings allow remote or hybrid schedules, and some positions may offer relocation or visa sponsorship-verify on each job page.
JAX: How do I apply for JAX jobs on Alion?
Open the JAX job posting and click Apply or follow the employer’s link; sign in to save jobs and enable alerts for new listings.