PyTorch FX Jobs
Explore PyTorch FX roles on Inference Jobs and apply today.
3mo agoMO
Member of Technical Staff - ML Performance
Modal
New York, United States (On-site)$150K – $270K Yearly
3mo agoBA
Software Engineer - Model Performance
Baseten
San Francisco, California, United States (On-site)$150K – $250K Yearly
2mo agoNV
Senior Compiler Engineer, AI Inference Platforms
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
3mo agoPE
AI Infra Engineer (San Francisco)
Perplexity
San Francisco, California, United States (On-site)$210K – $385K Yearly
3mo agoOP
Research-Hardware Codesign Engineer
OpenAI
San Francisco, California, United States (Hybrid)$230K – $460K Yearly
4w agoNV
Senior Software Engineer, Profiling Services
NVIDIA
Tel Aviv-Yafo, Tel Aviv District, Israel (On-site)
2mo agoNV
Senior Machine Learning Applications and Compiler Engineer
NVIDIA
Cambridge, England, United Kingdom (Hybrid)
3w agoTM
Research Engineer, Infrastructure, Kernels
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
3w agoTM
Research Engineer, Infrastructure, RL Systems
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
3d agoSC
Tech Lead Manager- MLRE, ML Systems
Scale
San Francisco, California, United States (On-site)$264.8K – $331K Yearly
3w agoTM
Software Engineer, Supercomputing
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
3mo agoD-
Machine Learning Intern - Dynamic KV-Cache Modeling for Efficient LLM Inference
d-Matrix
Santa Clara, Ca, Ca, United States or Remote (California, United States)$30 – $59 Hourly
3mo agoCO
Audio Inference Engineer, Model Efficiency
Cohere
New York, United States or Remote (New York, United States + 3 more)
3mo agoD-
Software Engineer, Staff - SIMD Kernels
d-Matrix
Santa Clara, Ca, Ca, United States or Remote (United States)$190K – $300K Yearly
4w agoTA
LLM Inference Frameworks and Optimization Engineer
Together AI
San Francisco, California, United States (On-site)$160K – $230K Yearly
1mo agoNV
AI Inference Performance Engineer - New College Grad 2026
NVIDIA
Santa Clara, California, United States (On-site)$124K – $241.5K Yearly
2mo agoGR