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

Exclusive to wagey.gg Pro - Machine Learning Scientist, Generative Molecular Design (Material Discovery)13h ago

Pro members are already applying...

United States - Hybrid·$134k - $252k + Equity

In OfficeNAMidCloud ComputingArtificial Intelligence3D DesignerDigital DesignerTensorFlowPyTorchAWS

+3 more Pro jobsin the last 48h

Fractile - Developer Experience / DevEx2d ago

·Unknown - USA *·Equity

In Office NA Artificial Intelligence Semiconductors Developer Experience Engineer Developer Advocate Rust JAX C++PyTorch Python OpenCL CUDA

SandboxAQ - Machine Learning Scientist4d ago

·United States - Hybrid·$134k - $252k/year + Equity

In Office NA Artificial Intelligence Transportation Machine Learning Engineer Learning & Development Python PyTorch JAX Reporting MAU Documentation

SandboxAQ - Machine Learning Scientist – Generative Molecular Design4d ago

·United States - Hybrid·$134k - $252k/year + Equity

In Office NA Mid Cloud Computing Artificial Intelligence UX Researcher 3D Designer TensorFlow PyTorch AWS JAX Oracle

Arena Intelligence, Inc. - Machine Learning Scientist4d ago

·Bay Area, California, United States

In Office NA Artificial Intelligence Machine Learning Engineer JAX Python PyTorch TensorFlow Learning & Development

olix - Senior Compiler Engineer5d ago

·London, UK·Equity

In Office EMEA Senior Artificial Intelligence Telecommunications Senior Software Engineer TensorFlow C++Coaching PyTorch JAX

10a Labs - Machine Learning Engineer5d ago

·Remote - USA *·$130k - $200k/year

Remote NA Mid Artificial Intelligence Machine Learning Engineer JAX Python PyTorch

Reflection - Member of Technical Staff1w ago

·San Francisco, California, United States·Equity

In Office NA Staff Artificial Intelligence Oil & Gas Staff Engineer JAX Python PyTorch Data Quality

Blue Rose Research - Machine Learning Lead (LLM)1w ago

·Remote·$165k - $210k/year

Remote WW Staff Artificial Intelligence Data Analytics Tech Lead SQL JAX Python PyTorch Team Leadership

Achira - Machine Learning Research Engineer (MLRE)1w ago

·San Francisco, California, United States - Hybrid·$165k - $259k/year + Equity

In Office NA Artificial Intelligence Nonprofit Machine Learning Engineer Research Engineer Documentation GitHub PyTorch JAX Observable

Lila Sciences - Senior / Staff Machine Learning Engineer, Applied AI1w ago

·Cambridge, MA USA; San Francisco, CA USA·$180k - $298k/year + Equity

In Office NA Staff Artificial Intelligence Machine Learning Engineer JAX Python PyTorch TensorFlow Ray

Reddit - Principal Data Scientist, User Ecosystem1w ago

·Remote - USA·$313k - $313k/year + Equity

Remote NA Principal Insurance Artificial Intelligence Data Scientist Principal Employee Relations PyTorch JAX E-learning Development Mentoring

mistral.ai - Applied Scientist - Research Engineer (Formal Methods), EMEA1w ago

·Paris

In Office EMEA Artificial Intelligence Applied Scientist Research Engineer PyTorch JAX Python

hirehire - RL Engineer | Humanoid1w ago

·London

In Office EMEA Mid Artificial Intelligence Robotics ZK Engineer Python PyTorch JAX Ray

hirehire - Robot Learning Engineer | Humanoid1w ago

·London

In Office EMEA Mid Robotics AI Engineer Postdoc Close PyTorch JAX

hirehire - VLA Pre-training & World Models | Humanoid1w ago

·London

In Office EMEA Mid Artificial Intelligence Robotics Training Specialist Transformers Python PyTorch JAX

Graphcore - Graduate Software Engineer, AI/ ML Test Systems 20271w ago

·Bristol, UK·Equity

In Office EMEA Junior Artificial Intelligence Data Analytics Software Engineer AI Engineer Python C++Linux Triton PyTorch JAX TensorFlow MLOps

Graphcore - Graduate Machine Learning Engineer 20271w ago

·Bristol, UK·Equity

In Office EMEA Junior Artificial Intelligence Data Analytics Machine Learning Engineer PyTorch JAX

Graphcore Early Careers - Graduate Software Engineer, AI/ ML Test Systems 20271w ago

·Bristol, UK·Equity

In Office EMEA Junior Artificial Intelligence Data Analytics Software Engineer AI Engineer Python C++Linux Triton PyTorch JAX TensorFlow MLOps

Graphcore Early Careers - Graduate Machine Learning Engineer 20271w ago

·Bristol, UK·Equity

In Office EMEA Junior Artificial Intelligence Data Analytics Machine Learning Engineer PyTorch JAX

lancedb - Senior Software Engineer2w ago

·Americas or APAC

In Office WW Senior Senior Software Engineer Ray JAX C++Java Scala Rust

Clariti Cloud Inc. - Senior Applied AI Engineer2w ago

·Remote - CANADA (Remote)·$150k - $180k/year

Remote NA Senior Cloud Computing Artificial Intelligence AI Engineer JAX Git Python MLOps AWS

cohere - Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)2w ago

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

Remote EMEA Staff Mental Health Artificial Intelligence Training Specialist JAX Data Quality Cody

Axiom Math - Member of Technical Staff3w ago

·Palo Alto, California, United States

In Office NA Staff Nonprofit Staff Engineer Python JAX Coaching

Graphcore - Senior Machine Learning Engineer (Large Systems)3w ago

·Bristol, UK - Hybrid·Equity

In Office EMEA Senior Insurance Cloud Computing Machine Learning Engineer JAX Reporting C++Python MLOps

ASTRABRIGHT ANALYTICS & KNOWLEDGE ALLIANCE, INC - Machine Learning Engineer3w ago

·and teams in Paris, Singapore·$175k - $275k/year + Equity

In Office APAC Mid Cloud Computing Artificial Intelligence Machine Learning Engineer JAX Python Kubernetes MLflow Airflow

PostEra - Machine Learning Researcher3w ago

·Remote - USA *·Equity

Remote NA Artificial Intelligence Nonprofit Machine Learning Engineer Data Scientist Ambassador Training Development Learning & Development Python JAX Cross-functional Collaboration

Menlo - Researcher, World Models3w ago

·Palo Alto, California, United States

In Office NA Robotics Research Scientist JAX

Airwallex - Senior Software Engineer, Machine Learning Platform3w ago

·Singapore

In Office APAC Senior Cloud Computing Artificial Intelligence Senior Software Engineer Machine Learning Engineer Java C++Python MLOps JAX

Trace - Machine Learning Engineer, Applied3w ago

·United States·$185k - $245k/year

In Office NA Mid Artificial Intelligence Aerospace Machine Learning Engineer Head of Engineering JAX Python Move

Achira - Machine Learning Research Engineer (MLRE)3w ago

·Remote - San Francisco, California, United States·$165k - $259k/year + Equity

Remote NA Artificial Intelligence Machine Learning Engineer Research Engineer JAX CUDA Triton Warp Observable

Brave - AI Security Researcher/Research Engineer3w ago

·North America, EU, UK - Hybrid

In Office NA Artificial Intelligence Research Engineer Security Engineer JAX Python Training Development

Flagship Pioneering, Inc. - Senior Machine Learning Scientist3w ago

·Cambridge, UK - Hybrid

In Office EMEA Senior Pharmaceuticals Biotechnology Senior Data Scientist Senior Researcher Python JAX Talent Acquisition hypothesis

AssemblyAI - Senior Research Engineer3w ago

·Remote·$600k - $600k/year

Remote WW Senior Artificial Intelligence Senior Researcher Research Engineer JAX Flax C++Rust Python

Charles River Analytics - Scientist3w ago

·Cambridge, MA - Hybrid - USA *·$120k - $200k/year

In Office NA Mid Artificial Intelligence Robotics Research Scientist JAX Linear Reporting Linux Python

Flagship Pioneering, Inc. - Senior Director / Vice President, Machine Learning (Generative Chemistry)3w ago

·Cambridge, MA USA·$279k - $413k/year

In Office NA Vp Pharmaceuticals Biotechnology President Executive Director Team Management Python JAX Prefect Decision Making

anyone-ai - GPU Kernel Engineer – CUDA, Triton & Accelerator Performance4w ago

·Remote - Uruguay, Chile, Ecuador...·$135k - $135k/year

Remote LATAM Mid Banking Cloud Computing Artificial Intelligence Firmware Engineer Graphics Engineer Triton CUDA Technical Writing AWS JAX

Airwallex - Staff AI Engineer4w ago

·San Francisco, California, United States·$182k - $310k/year + Equity

In Office NA Staff Banking Payments AI Engineer Staff Engineer JAX Python

cantina - Research Intern4w ago

·Singapore

In Office APAC Intern Artificial Intelligence Research Intern JAX Python Reporting Temporal

IMC - Machine Learning Engineer4w ago

·Amsterdam, Netherlands; London, United Kingdom

In Office EMEA Senior Artificial Intelligence Data Analytics Machine Learning Engineer JAX Python Training Development CUDA C++

IMC - Graduate Machine Learning Researcher4w ago

·London, United Kingdom

In Office EMEA Junior Artificial Intelligence Higher Education Machine Learning Engineer Graduate Trainee PyTorch Python JAX Transformers Risk Management

Discord - Staff Software Engineer, Machine Learning (Consumer Revenue)4w ago

·Remote - USA·$272k - $340k/year + Equity

Remote NA Staff Artificial Intelligence Gaming Software Engineer Staff Engineer JAX Python Product Marketing Ray Discord

Mistral - Research Engineer, Forge4w ago

·Paris, Ile-de-France, France

In Office EMEA Artificial Intelligence Research Engineer Kubernetes Ray Documentation Python JAX

Airwallex - Senior AI Engineer (Risk & Payments)4w ago

·Singapore

In Office APAC Senior Banking Payments AI Engineer Recruiter Python JAX Governance

Physical Superintelligence - Member of Technical Staff, AI Research4w ago

·Boston, Massachusetts, United States

In Office NA Staff Artificial Intelligence Gaming Staff Scientist Training Development JAX Ray Transformers Go

AMI - Scientist/Interns4w ago

·Paris, France

In Office EMEA Artificial Intelligence Staff Scientist Research Intern JAX Python

Persona AI Inc - Robotics Machine Learning Engineer4w ago

·Houston, Texas, United States·Equity

In Office NA Mid Cloud Computing Artificial Intelligence Machine Learning Engineer Learning & Development PyTorch JAX AWS GCP

Causal Labs - Member of Technical Staff — Inference Infrastructure4w ago

·San Francisco, California, United States

In Office NA Staff Staff Engineer JAX Kubernetes Ray Triton

Causal Labs - Member of Technical Staff — Training Infrastructure4w ago

·San Francisco, California, United States

In Office NA Staff Artificial Intelligence Nonprofit Training Specialist JAX PyTorch

mistral.ai - Applied Scientist/Research Engineer, EMEA3mo ago

·Paris, Ile-de-France, France

In Office EMEA Cybersecurity Artificial Intelligence Applied Scientist Research Engineer Founder JAX Python

Get Started Free

No credit card. Takes 10 seconds.