Remote Tech JAX Jobs

94 remote jobs

Upload My Resume

Drag and drop resume here, or click to browse

Supports PDF, DOCX, DOC, RTF, and TXT

Cohere - Senior Member of Technical Staff, Multimodal AI3mo ago

·Remote - United States, France, Canada·$156k - $156k/year

humanoid - VLA Pre-training Engineer (Deep Learning)3mo ago

·London, United Kingdom·Equity

In Office EMEA Mid Artificial Intelligence Robotics Junior Engineer Python JAX MLOps

humanoid - Reinforcement Learning Engineer - Manipulation3mo ago

·London, United Kingdom·Equity

In Office EMEA Mid Artificial Intelligence Robotics AI Engineer Python JAX

humanoid - Deep Learning Engineer, World Models3mo ago

·London, United Kingdom·Equity

In Office EMEA Robotics Oil & Gas Machine Learning Engineer Learning & Development Python Transformers JAX

sandboxaq - ML Research Engineer, AI for Life Sciences3mo ago

·United Kingdom·£71k - £126k/year + Equity

In Office EMEA Senior Pharmaceuticals Life Sciences Artificial Intelligence ML Engineer Research Engineer Staff Development Training Development Weights & Biases Python JAX

axiombio - ML Researcher3mo ago

·San Francisco, California, USA

In Office NA Artificial Intelligence Nonprofit User Researcher Technical Writing Customer Training Learning & Development Pandas JAX

Graphcore - AI Research Engineer3mo ago

·Cambridge, UK

In Office EMEA Artificial Intelligence Semiconductors Research Engineer AI Engineer Python Reporting JAX CUDA C++

Inferact - Member of Technical Staff, TPU Performance Engineering3mo ago

·Singapore·$155k - $310k/year + Equity

In Office APAC Staff Artificial Intelligence Staff Engineer JAX

PhysicsX - Applied Scientist - Senior / Principal3mo ago

·Singapore·Equity

In Office APAC Principal Artificial Intelligence Applied Scientist Principal Technical Writing Learning & Development Python Pandas JAX

ami - Engineer - Singapore4mo ago

·Singapore

In Office EMEA Artificial Intelligence Build Engineer JAX Python

AvePoint - AI Engineer (Managed Services)4mo ago

·Unknown - Asia-Pacific *

In Office APAC Mid Artificial Intelligence Higher Education Cloud Computing AI Engineer QA Engineer Software Engineer Junior Software Engineer Teacher JAX Python FastAPI Flask Entity Framework

Menlo - Robotics AI Engineer4mo ago

·Ho Chi Minh City, Vietnam

In Office APAC Robotics AI Engineer JAX Python Close

adaption - Modelling Resident (8-Month Contract)4mo ago

·Remote - United States, Mexico, France... - Latin America *

Remote LATAM Graduate Student Python JAX React

inherent - Member of Technical Staff (Infrastructure Engineer, Training and Inference Systems)4mo ago

·London, England, United Kingdom

In Office EMEA Staff Artificial Intelligence Infrastructure Engineer Staff Engineer JAX Python Rust C++CUDA

inherent - Member of Technical Staff (Post Training)4mo ago

·London, England, United Kingdom

In Office EMEA Staff Artificial Intelligence Staff Engineer Python JAX

adaption - Senior Research Scientist5mo ago

·United States

In Office NA Senior Research Scientist Senior Researcher JAX Python

Cohere - Machine Learning Intern/Co-op (Winter 2027)5mo ago

·Remote - United States, European Union, United Kingdom, Canada

Remote EMEA Intern Mental Health Artificial Intelligence Co-op Intern JAX Python TensorFlow Training Development Transformers

Cohere - Research Internship (Fall / Winter 2026)6mo ago

·Remote - United Kingdom, European Union, United States, Canada

Remote EMEA Intern Mental Health Artificial Intelligence Research Intern Associate C++Lua JAX Python PyTorch

bjakcareer - Member of Technical Staff, Machine Learning6mo ago

·London, Greater London, United Kingdom

In Office EMEA Staff Artificial Intelligence Machine Learning Engineer JAX Coaching

bjakcareer - Principal Machine Learning Engineer6mo ago

·London, Greater London, United Kingdom

In Office EMEA Principal Artificial Intelligence Machine Learning Engineer Principal JAX Ray

spotify - Senior Staff Machine Learning Engineer, Content Platform6mo ago

·London / Stockholm

Remote EMEA Staff Artificial Intelligence Oil & Gas Machine Learning Engineer Customer Success Engineer JAX Data Quality

Figma - AI Applied Scientist6mo ago

·Remote - USA·$153k - $376k/year + Equity

Remote NA Mid Mental Health Artificial Intelligence Applied Scientist Recruiter C++Python Java JAX Learning & Development

Anthropic - Research Engineer, Machine Learning (RL Velocity)6mo ago

·Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY - Hybrid·$500k - $500k/year + Equity

In Office NA Banking Artificial Intelligence Research Engineer Machine Learning Engineer JAX

Anthropic - Research Engineer, Machine Learning (RL Velocity)6mo ago

·London, UK - Hybrid·£370k/year/year + Equity

In Office EMEA Banking Artificial Intelligence Research Engineer Machine Learning Engineer JAX

webAI - Senior Forward Deployed AI Engineer6mo ago

·Washington, District of Columbia, United States

In Office NA Senior Artificial Intelligence Nonprofit AI Engineer Advisor ONNX JAX Full Stack

Astera Institute - Research Scientist: Energy Based Models6mo ago

·Emerybille, California, United States·$150k - $300k/year

In Office NA Genomics Research Scientist JAX

zencore - Principal Architect, AI/ML6mo ago

·Remote - Anywhere - Europe *

Remote EMEA Principal Cloud Computing Artificial Intelligence Principal Principal Engineer Coaching Business Development Claude Gemini JAX

bjakcareer - Applied AI Engineer6mo ago

·Singapore, Orchard Road

In Office APAC Artificial Intelligence Nonprofit AI Engineer JAX

hcompany - Research Engineer / Scientist, Post-training & Reinforcement Learning6mo ago

·United Kingdom - Hybrid

In Office EMEA Staff Artificial Intelligence Research Engineer Staff Engineer Machine Learning Engineer Python JAX Rust PyTorch Git

beyondmath - AI Researcher6mo ago

·London, England, United Kingdom - Hybrid

In Office EMEA Senior Artificial Intelligence AI Engineer JAX C++CUDA Transformers Cross-functional Collaboration

hcompany - Research Engineer, Model Inference & Serving6mo ago

·United Kingdom - Hybrid

In Office EMEA Staff Artificial Intelligence Materials Research Engineer Staff Engineer Rust C++Python Go JAX

perplexity - Internship - Search Machine Learning Engineer6mo ago

·London, UK, United Kingdom

In Office EMEA Intern Artificial Intelligence Nonprofit Machine Learning Engineer Engineering Intern Python JAX

perplexity - Member of Technical Staff (AI Inference Engineer)6mo ago

·London, UK, United Kingdom·Equity

In Office EMEA Mid Artificial Intelligence AI Engineer Staff Engineer JAX CUDA Triton Rust Python

Runway - Member of Technical Staff, Research Engineer6mo ago

·United States·$270k - $370k/year

In Office NA Staff Artificial Intelligence Research Engineer Staff Engineer JAX

cohere - Senior Member of Technical Staff, Safety and Security for Agents7mo ago

·Remote - United States, United Kingdom, Canada, France - USA *·$205k - $380k/year

Remote EMEA Staff Mental Health Artificial Intelligence Security Management Specialist Director of Security JAX Python

Zencore - Principal Architect, AI/ML8mo ago

·Remote - EMEA (Remote)

Remote EMEA Principal Cloud Computing Artificial Intelligence Principal Principal Engineer Coaching Claude Gemini Business Development JAX

Bjak - Principal Machine Learning Engineer8mo ago

·Beijing, China

In Office APAC Principal Artificial Intelligence Machine Learning Engineer Principal JAX Ray

Anthropic - Research Engineer, Machine Learning (Reinforcement Learning)8mo ago

·London, UK - Hybrid·£260k - £630k/year/year + Equity

In Office EMEA Banking Artificial Intelligence Research Engineer Lawyer Machine Learning Engineer Python JAX

Isomorphic Labs - Research Scientist (Machine Learning), London8mo ago

·London - Hybrid

In Office EMEA Artificial Intelligence Robotics Research Scientist Linear Training Development Reporting JAX Pandas

adaption - Research Scientist / Research Engineer8mo ago

·Remote - United States, Canada, United Kingdom, India

Remote NA Mid Research Engineer Research Scientist JAX Python React

Point72 - Quantitative Researcher8mo ago

·New York, London, or Hong Kong·Equity

In Office APAC Artificial Intelligence Quantitative Researcher JAX Python

featherlessai - AI Researcher – Multilingual Data9mo ago

·Remote - (world) - USA *·Equity

Remote NA Artificial Intelligence Data Analyst Data Quality Python JAX

featherlessai - AI Researcher — AI Architecture Research9mo ago

·Remote (world) - Hybrid·Equity

In Office WW Artificial Intelligence Architecture Research Engineer JAX

Featherless AI - Machine Learning Engineer — Distillation9mo ago

·Remote - (world)·Equity

Remote EMEA Artificial Intelligence Nonprofit Machine Learning Engineer Teacher JAX

Featherless AI - Machine Learning Engineer — AI Architecture Research9mo ago

·Remote (world) - Hybrid·Equity

In Office WW Artificial Intelligence Machine Learning Engineer AI Engineer Customer Training JAX

Get Started Free

No credit card. Takes 10 seconds.