- Founding Research Engineer at Flower — Anywhere
- Research Engineer / Research Scientist at Ideogram — Canada
- Applied Research Engineer, Motion Planning at Intrinsic — United States
- Lead Machine Learning Engineer at InstaDeep — United Kingdom
- Lead Machine Learning Engineer at Instadeep — United Kingdom
- AI Research Intern at Contextual AI — United States
- Machine Learning Community Operations Engineer at Clarifai — India
- AI Performance Optimization Engineer at Lightning AI — United States
- AI Engineer - Reasoning at xAI — United States
- Senior Research Engineer at AssemblyAI — Anywhere
- AI Engineer Reasoning at xAI — United States
- Senior AI Engineer at Atomic AI — United States
- Principal AI Solutions Software Engineer, AI Systems at SambaNova Systems — United States
- Machine Learning Researcher at Regrello — Anywhere
- Machine Learning Researcher at Modern Intelligence — United States
- Software Engineer - Machine Learning Infrastructure at Moloco — United States
- Software Engineer - Machine Learning Infrastructure at Moloco — United States
- Research Scientist at Luma AI — United States
- Software Engineer - Machine Learning Infrastructure at Moloco — United States
- Jr Fullstack Software Engineer - Java at Accurate Background — India
Find
jax jobs.
Browse 20 open Jax positions. Find remote and on-site Jax jobs at top companies hiring now.
Flower
Founding Research Engineer
Remote
Full Time
#Engineering
#Artificial Intelligence
#LLM
#Training
#Optimization
#Experimental Design
#PyTorch
#Jax
#Linux
#Docker
I
Ideogram
Research Engineer / Research Scientist
On-site
Full Time
#Machine Learning
#AI
#Artificial Intelligence
#Jax
#PyTorch
#TensorFlow
#Diffusion Models
#Deep Learning
#Kubernetes
#Docker
#CUDA
I
Intrinsic
Applied Research Engineer, Motion Planning
200k - 230k USD
On-site
Full Time
#Robotics
#Artificial Intelligence
#Engineering
#Planning
#Machine Learning
#C++
#Python
#Jax
#Reinforcement Learning
InstaDeep
Lead Machine Learning Engineer
Hybrid
Full Time
#Research
#Python
#C C++
#Linux Systems
#Performance Analysis
#Machine Learning
#Jax
#TensorFlow
#PyTorch
#CUDA
Instadeep
Lead Machine Learning Engineer
Hybrid
Full Time
#Research
#Python
#C C++
#Linux Systems
#Performance Analysis
#Machine Learning
#Jax
#TensorFlow
#PyTorch
#CUDA
Contextual AI
AI Research Intern
110k - 120k USD
On-site
Full Time
#Artificial Intelligence
#Machine Learning
#Engineering
#Python
#Rust
#C++
#Jax
#PyTorch
#Research Analysis
#Problem Solving
#Communication
Clarifai
Machine Learning Community Operations Engineer
On-site
Full Time
#AI
#Computer Vision
#Machine Learning
#Deep Learning
#Python
#PyTorch
#TensorFlow
#Jax
#Cloud Platforms
#Git
#Engineering
Lightning AI
AI Performance Optimization Engineer
On-site
Full Time
#Engineering
#Machine Learning
#AI Development
#PyTorch
#Jax
#TensorFlow
#CUDA
#Compiler Development
#Distributed Training
#Optimization
xAI
AI Engineer - Reasoning
180k - 440k USD
On-site
Full Time
#Engineering
#Python
#Jax
#Rust
#Reinforcement Learning
#Distributed Systems
AssemblyAI
Senior Research Engineer
185k - 260k USD
Remote
Full Time
#Engineering
#AI
#Jax
#Distributed Training
#Python
#Hardware
#Deep Learning
#Transformers
#Learning
xAI
AI Engineer Reasoning
180k - 440k USD
On-site
Full Time
#Engineering
#Python
#Jax
#Rust
#Distributed Systems
#Reinforcement Learning
Atomic AI
Senior AI Engineer
190k - 220k USD
Hybrid
Full Time
#Engineering
#Biotech
#AI
#Python
#Kubernetes
#Docker
#AWS
#MLOps
#Deep Learning
#Git
#CI CD
#PyTorch
#Jax
SambaNova Systems
Principal AI Solutions Software Engineer, AI Systems
165k - 190k USD
On-site
Full Time
#Software Engineering
#Python
#C++
#Data Structures
#Algorithms
#Machine Learning
#PyTorch
#TensorFlow
#Jax
#Deep Learning
#Compiler
Regrello
Machine Learning Researcher
Remote
Full Time
#Engineering
#AI
#Supply Chain
#Python
#PyTorch
#TensorFlow
#Jax
#Deep Learning
#Network
#Natural Language Processing
#Machine Learning
Modern Intelligence
Machine Learning Researcher
On-site
Full Time
#Engineering
#Machine Learning
#Artificial Intelligence
#PyTorch
#TensorFlow
#Jax
#Data Science
#Computer Vision
#Object Detection
#Learning
Moloco
Software Engineer - Machine Learning Infrastructure
133k - 200k USD
On-site
Full Time
#Engineering
#Machine Learning
#Infrastructure
#Python
#Java
#C#
#Go
#C++
#Jax
#Distributed Systems
Moloco
Software Engineer - Machine Learning Infrastructure
133k - 200k USD
On-site
Full Time
#Engineering
#Machine Learning
#Advertising
#Python
#Java
#C#
#Go
#C++
#PyTorch
#Jax
#GCP
#AWS
#Azure
Luma AI
Research Scientist
On-site
Full Time
#AI
#Research
#Engineering
#PyTorch
#Jax
#Reinforcement Learning
#Data
#Distributed Systems
Moloco
Software Engineer - Machine Learning Infrastructure
133k - 200k USD
On-site
Full Time
#Engineering
#Machine Learning
#Infrastructure
#Python
#Java
#C#
#Go
#C++
#PyTorch
#Jax
#GCP
#AWS
#Azure
Accurate Background
Jr Fullstack Software Engineer - Java
Hybrid
Full Time
#Technology
#Engineering
#Java
#Spring
#J2EE Technologies
#RESTful Web Services
#XML
#Jax
#JSON
#Tomcat
#JBoss