JAX Jobs
Browse 137 JAX jobs on Inference Jobs.
81-100 of 137 jobs
3wXA
Member of Technical Staff, Grok Imagine
xAI
Palo Alto, California, United States (On-site)$180k – $440k Yearly
4wNV
Deep Learning Software Engineer, FlashInfer - New College Grad 2025
NVIDIA
Santa Clara, California, United States (On-site)$108k – $195.5k Yearly
2wPE
Internship - Search Machine Learning Engineer (Belgrade)
Perplexity
Belgrade, Belgrade, Serbia (On-site)
11hNV
Senior Software Engineer - Deep Learning Compiler Verification and Infrastructure
NVIDIA
Santa Clara, California, United States (On-site)$140k – $224.3k Yearly
2wNV
Senior Deep Learning Compiler Engineer - PyTorch
NVIDIA
Berlin, Berlin, Germany (On-site)zł 292.5k – zł 507k Yearly
3dNV
Senior AI Compiler Engineer, MLIR
NVIDIA
Santa Clara, California, United States (On-site)$152k – $241.5k Yearly
5dXA
5dXA
Member of Technical Staff, Video Generation - Audio
xAI
San Francisco, California, United States (On-site)$180k – $440k Yearly
2wNV
Deep Learning Compiler Verification and Infra Development Intern - 2026
NVIDIA
Shanghai, Shanghai, China (On-site)
1wOP
Research-Hardware Codesign Engineer
OpenAI
San Francisco, California, United States (Hybrid)$230k – $460k Yearly
1wSC
Machine Learning Research Engineer, GenAI Applied ML
Scale
San Francisco, California, United States (On-site)$176k – $220k Yearly
3dNV
Senior Compiler Engineer, AI Inference Performance
NVIDIA
Santa Clara, California, United States (On-site)$152k – $241.5k Yearly
2wNV
High-Performance LLM Training Engineer - New College Grad 2026
NVIDIA
Santa Clara, California, United States (On-site)$124k – $195.5k Yearly
1wNV
Applied Atomistic Modeling Researcher, Drug Discovery
NVIDIA
Santa Clara, California, United States (On-site)$192k – $304.8k Yearly
5dMA