Sovia

46 Jobs Requiring JAX

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

Member of Technical Staff - Voice ModelxAI
Low Ghost Risk
Alto, CA$150K/yr - $450K/yr1 weeks agoActive
Mid, Senior Level
full-time
Python
Spark
Ray
Speech-language model pre-training
Speech-language model post-training
+9 more
Research Engineer - Computer Vision MLSesame
Low Ghost Risk
Francisco, CA$190K/yr - $320K/yr1 weeks agoActive
Mid, Senior Level
full-time
Machine Learning
Computer Vision
3D Computer Vision
Deep Learning Frameworks
PyTorch
+3 more
Scientific Python Developer (STEM) | RemoteCrossing Hurdles
Low Ghost Risk
Remote$70/hr - $90/hr1 weeks agoActive
Mid Level
contract
Python
Scientific computing libraries
NumPy
SciPy
Matplotlib
+9 more
Research Engineer – Python (STEM) | $90/hr RemoteCrossing Hurdles
Low Ghost Risk
Remote$70/hr - $90/hr1 weeks agoActive
Mid Level
contract
Python
Scientific computing libraries
NumPy
SciPy
Matplotlib
+12 more
Senior Machine Learning ScientistExpedia Group
Low Ghost Risk
Jose, CA$187K/yr - $262K/yr1 weeks agoActive
full-time
Agentic AI systems
Multi-step AI systems
ReAct
Tool-calling agents
Multi-agent pipelines
+19 more
AI Research Engineer – Scientific Computing | $90/hr RemoteCrossing Hurdles
Low Ghost Risk
Remote$70/hr - $90/hr1 weeks agoActive
Mid Level
contract
Python
Scientific computing libraries
NumPy
SciPy
Matplotlib
+12 more
Member of Technical Staff (ML)Reka AI
Low Ghost Risk
United States (Remote)$62K/yr - $104K/yr1 weeks agoActive
Mid Level
full-time
Deep Learning
Machine Learning Algorithms
PyTorch
JAX
Model Training
+3 more
Artificial Intelligence / Machine Learning Data EngineerMAG Aerospace
Low Ghost Risk
Fairfax, VA$165K/yr - $231K/yr1 weeks agoActive
Senior Level
full-time
Machine Learning
Artificial Intelligence
Data Engineering
Python
PyTorch
+25 more
Senior Member of Technical Staff, Multimodal AICohere
Low Ghost Risk
United States (Remote)$155K/yr - $165K/yr1 weeks agoActive
Senior Level
full-time
Software engineering
Python
JAX
PyTorch
TensorFlow
+7 more
Member of Technical Staff, Research Engineer (GPU Performance)Odyssey Logistics
Low Ghost Risk
United States (Remote)$270K/yr - $370K/yr1 weeks agoActive
Mid, Senior Level
full-time
GPU kernel development
CUDA
Triton
CUTLASS
Distributed systems
+13 more
Software Engineer III, AI/ML GenAI, GCP, PerformanceGoogle
Low Ghost Risk
Sunnyvale, CA$147K/yr - $211K/yr1 weeks agoActive
Entry, Mid Level
full-time
Python
C++
Java
Machine learning frameworks
TensorFlow
+15 more
Software Engineer III, AI/ML GenAI, GCP, PerformanceGoogle
Low Ghost Risk
Kirkland, WA$147K/yr - $211K/yr2 weeks agoActive
Entry, Mid Level
full-time
Python
C++
Java
Machine learning frameworks
TensorFlow
+15 more
Senior Machine Learning Engineer, AI Product and PlatformBetterUp
Low Ghost Risk
Engineer, AI$200K/yr - $250K/yr2 weeks agoActive
Mid, Senior Level
full-time
Machine Learning Systems
Large Language Model (LLM) Fine-tuning
Prompt Engineering
Retrieval-Augmented Generation (RAG) Pipelines
Python
+9 more
Computational Scientist – Python | RemoteCrossing Hurdles
Low Ghost Risk
Remote$70/hr - $90/hr2 weeks agoActive
Mid Level
contract
Python
Scientific computing libraries
NumPy
SciPy
Matplotlib
+12 more
Robotics Engineer, ManipulationSkild AI
Low Ghost Risk
Francisco, CA$100K/yr - $300K/yr2 weeks agoActive
Entry, Mid Level
full-time
Robotic manipulation systems
Robot software deployment
Robotics controllers
Kinematics
Dynamics
+23 more
Senior Solutions Architect, Financial Services BankingNVIDIA
Low Ghost Risk
United States (Remote)$180K/yr - $280K/yr2 weeks agoActive
full-time
Machine Learning
Deep Learning
Python
C++
TensorFlow
+17 more
Research EngineerSkild AI
Low Ghost Risk
Francisco, CA$100K/yr - $300K/yr2 weeks agoActive
Entry, Mid, Senior Level
full-time
Python
Deep learning libraries
PyTorch
TensorFlow
JAX
+7 more
Senior Machine Learning EngineerMicrosoft
Low Ghost Risk
Redmond, WA$120K/yr - $235K/yr2 weeks agoActive
Mid, Senior Level
full-time
Machine Learning
Large Language Models
Multimodal Systems
Reinforcement Learning
Agentic AI Architectures
+15 more
Biology Expert | RemoteCrossing Hurdles
Low Ghost Risk
Remote$70/hr - $120/hr2 weeks agoActive
Mid Level
contract
Python
Scientific Computing
NumPy
SciPy
Matplotlib
+6 more
Full Stack Engineer - AI Accelerator Open Source EcosystemSemiAnalysis
Low Ghost Risk
United States (Remote)$110K/yr - $120K/yr2 weeks agoActive
Entry, Mid, Senior Level
full-time
React
Next.js
TypeScript
SQL
PostgreSQL schema design
+23 more

People Also Search For

Browse All SkillsBrowse Categories