Sovia

46 Jobs Requiring JAX

Browse all positions that list JAX as a required or preferred skill

Senior Deep Learning Software Engineer, LLM PerformanceNVIDIA
Low Ghost Risk
Engineer, LL (Remote)$184K/yr - $288K/yr4 weeks agoActive
full-time
Deep Learning
Python
C++
Software Design
Software Engineering
+8 more
Staff Machine Learning Engineer – VLM/LLM EvaluationWaymo
Low Ghost Risk
View, CA$238K/yr - $302K/yr1 months agoAging
full-time
Machine Learning
Deep Learning
Reinforcement Learning
Generative Modeling
Python
+8 more
System Software Engineer - AIDelos Data
Moderate Ghost Risk
United States (Remote)$140K/yr - $200K/yr2 months agoStale
Mid Level
full-time
C++
Python
Systems Programming
Linux
GPU Programming
+7 more
AI Researcher — AI Architecture ResearchFeatherless AI
Moderate Ghost Risk
United States (Remote)$160K/yr - $250K/yr2 months agoStale
Mid, Senior Level
full-time
Machine Learning
Neural Network Architectures
Sequence Models
Attention Mechanisms
Training Dynamics
+6 more
Machine Learning Engineer — AI Architecture ResearchFeatherless AI
Moderate Ghost Risk
United States (Remote)$118K/yr - $158K/yr2 months agoStale
Mid, Senior Level
full-time
Machine Learning
Deep Learning
Neural Network Architectures
Attention Mechanisms
Recurrent Neural Networks
+8 more
Machine Learning Engineer — DistillationFeatherless AI
Moderate Ghost Risk
United States (Remote)$118K/yr - $158K/yr2 months agoStale
Mid, Senior Level
full-time
Machine Learning
Deep Learning
Model Distillation
PyTorch
JAX
+7 more

People Also Search For

Browse All SkillsBrowse Categories