Remote Tech JAX Jobs

96 remote jobs

Upload My Resume

Drag and drop resume here, or click to browse

Supports PDF, DOCX, DOC, RTF, and TXT

Cohere - Machine Learning Intern/Co-op (Fall, 2026)4mo ago

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

SmarterDx - Staff Machine Learning Research Scientist4mo ago

·Remote - USA·$220k - $260k/year

Remote NA Staff Cloud Computing Artificial Intelligence Research Scientist Staff Scientist JAX Python Transformers Hugging Face AWS

Cohere - Research Internship (Fall / Winter 2026)4mo 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 Learning4mo ago

·London, Greater London, United Kingdom

In Office EMEA Staff Artificial Intelligence Machine Learning Engineer JAX Coaching

bjakcareer - Staff Machine Learning Engineer4mo ago

·London, Greater London, United Kingdom

In Office EMEA Staff Artificial Intelligence Machine Learning Engineer JAX

bjakcareer - Principal Machine Learning Engineer4mo 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 Platform4mo ago

·London / Stockholm

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

Figma - AI Applied Scientist4mo ago

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

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

PhysicsX - Principal Research Scientist4mo ago

·London, United Kingdom·Equity

In Office EMEA Principal Artificial Intelligence Research Scientist Principal Technical Writing Learning & Development CUDA Python JAX

Anthropic - Research Engineer, Machine Learning (RL Velocity)4mo 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)4mo 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 Engineer4mo ago

·Austin, Texas, United States

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

Astera Institute - Research Scientist: Energy Based Models5mo ago

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

In Office NA Genomics Research Scientist JAX

yubo - Staff ML Engineer5mo ago

·Paris, Ile de France, France - Hybrid·Equity

Remote EMEA Staff Artificial Intelligence ML Engineer Staff Engineer Talent Acquisition Manager Python KPI Tracking JAX MLOps Case Management

zencore - Principal Architect, AI/ML5mo ago

·Remote - Anywhere - Europe *

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

bjakcareer - Applied AI Engineer5mo ago

·Singapore, Orchard Road

In Office APAC Artificial Intelligence Nonprofit AI Engineer JAX

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

·United Kingdom - Hybrid

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

beyondmath - AI Researcher5mo ago

·London, England, United Kingdom - Hybrid

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

bjakcareer - Technical Lead, Machine Learning5mo ago

·Singapore, Orchard Road

In Office APAC Staff Artificial Intelligence Tech Lead Machine Learning Engineer JAX

Bjak - Principal Machine Learning Engineer5mo ago

·Singapore, Orchard Road

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

bjakcareer - VP of Research, Machine Learning5mo ago

·Singapore, Orchard Road

In Office APAC Vp Artificial Intelligence VP of Engineering VP of Data JAX Close

telus-digital - AI Research Engineer5mo ago

·Remote - Brazil

Remote LATAM Mid Artificial Intelligence Research Engineer AI Engineer JAX Python Instructional Design DVC JSON

hcompany - Research Engineer, Model Inference & Serving5mo 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 Engineer5mo 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)5mo 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 Engineer5mo 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 Agents5mo ago

·Remote - United States, United Kingdom, Canada, France - USA *

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

Runway - Member of Technical Staff, Machine Learning Datasets6mo ago

·Remote

Remote WW Staff Artificial Intelligence Oil & Gas Senior Data Scientist Documentation SQL Data Analysis JAX Kubernetes

Reddit - Staff Research Engineer, Pre-training Data6mo ago

·Remote - USA·$230k - $322k/year + Equity

Remote NA Staff Artificial Intelligence Data Analytics Staff Engineer Research Engineer Training Development Python Ray JAX Rust

Socure - PhD Research Intern – Graph Learning & Agentic AI for Fraud Detection6mo ago

·Carson City, nevada, United States - Hybrid·$52k - $73k/year

In Office NA Intern Artificial Intelligence Materials Research Intern Fraud Analyst Learning & Development Transformers JAX

EyeTell, Inc. - Senior AI Engineer6mo ago

·San Mateo, California, USA

In Office NA Senior Artificial Intelligence AI Engineer Senior Software Engineer JAX Python

Intrinsic - Robotics Software Engineer, Robot Learning and Manipulation6mo ago

·Singapore

In Office APAC Mid Robotics Software Engineer Machine Learning Engineer C++Python JAX

Spotify - Senior Staff Machine Learning Engineer6mo ago

·New York, NY·$265k - $378k/year

In Office NA Staff Artificial Intelligence Nonprofit Machine Learning Engineer Ambassador Java JAX Python Scala Team Leadership

Zencore - Principal Architect, AI/ML6mo ago

·Remote - EMEA (Remote)

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

perplexity - AI Inference Internship7mo ago

·London, UK, United Kingdom - Hybrid

In Office EMEA Intern Artificial Intelligence Intern Software Engineering Intern JAX CUDA Triton

Bjak - Principal Machine Learning Engineer7mo ago

·Beijing, China

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

Anthropic - Research Engineer, Machine Learning (Reinforcement Learning)7mo 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), London7mo ago

·London - Hybrid

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

adaption - Research Scientist / Research Engineer7mo ago

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

Remote NA Mid Research Engineer Research Scientist JAX Python React

Scale AI - Machine Learning Engineer, International Public Sector7mo ago

·Doha, Qatar; London, UK

In Office EMEA Senior Artificial Intelligence Public Sector Machine Learning Engineer C++Python TypeScript JavaScript JAX

Point72 - Quantitative Researcher7mo ago

·New York, London, or Hong Kong·Equity

In Office APAC Artificial Intelligence Quantitative Researcher JAX Python

Terra AI - Full Stack Engineer, Scientific Modeling Tools7mo ago

·Redwood City, California, United States

In Office NA Artificial Intelligence Renewable Energy Full Stack Engineer Python Full Stack Julia JAX Prefect

featherlessai - AI Researcher – Multilingual Data7mo ago

·Remote - (world) - USA *·Equity

Remote NA Artificial Intelligence Data Analyst Data Quality Python JAX

featherlessai - AI Researcher — AI Architecture Research7mo ago

·Remote (world) - Hybrid·Equity

In Office WW Artificial Intelligence Architecture Research Engineer JAX

Featherless AI - Machine Learning Engineer — Distillation7mo ago

·Remote - (world)·Equity

Remote EMEA Artificial Intelligence Nonprofit Machine Learning Engineer Teacher JAX

Featherless AI - Machine Learning Engineer — AI Architecture Research7mo 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.