2,015 pytorch jobs at 539 companies in California
2mo
Save
Mark Applied
Hide
Compiler Engineer (PyTorch / Triton / LLVM / MLIR)
Los Altos, California, United States
OnsiteFull Time
7+ YOEBachelor's or Master's in CS/CE, 7+ years compiler engineering experience, deep LLVM/MLIR knowledge, PyTorch and Triton experience, proficiency in C/C++, performance optimization and strong communication skills.
PyTorch, PyTorch Inductor, Dynamo, Triton, LLVM, MLIR, C/C++
2mo
Save
Mark Applied
Hide
Staff Compiler Engineer - PyTorch + Kernel DSLPLATE
San Jose, California, United States
$163k-$253k/yr OnsiteFull Time
10+ YOEBS with 10+ yrs, MS with 8+ yrs, or PhD with 5+ yrs; experience in Triton/Helion/MLIR/XLA/TVM/Inductor/IREE/CUTLASS; kernel DSL/IR design; MLIR; PyTorch backends; kernel autotuning; HPC/NUMA; open-source contributions.
Triton, Helion, MLIR, XLA, TVM, Inductor, IREE, CUTLASS, PyTorch, LLVM/MLIR
3d
Save
Mark Applied
Hide
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS)
McLean or Richmond or New York City or San Jose or Cambridge
$179k-$246k/yr OnsiteFull Time
6+ YOE2+ MgmtBachelor's degree, 6+ years designing data-intensive distributed solutions, 4+ years programming in Python, Scala, or Java, and 2+ years building and optimizing ML systems.
KServe, Kubernetes, PyTorch, TensorFlow, AWS, Python, Scala, Java, scikit-learn, Dask, Spark, Azure, Google Cloud Platform
3d
Save
Mark Applied
Hide
Lead Machine Learning Engineer (MLOps, KServe + building Kubernetes Clusters, PyTorch, TensorFlow on AWS)
McLean or Cambridge or San Jose or Richmond
$179k-$246k/yr OnsiteFull Time
6+ YOE2+ MgmtBachelor's degree, 6+ years designing data-intensive distributed systems, 4+ years programming in Python, Scala, or Java, and 2+ years building, scaling, and optimizing ML systems.
KServe, Kubernetes, PyTorch, TensorFlow, AWS, Python, Scala, Java, scikit-learn, Dask, Spark, Azure, Google Cloud Platform
3w
Save
Mark Applied
Hide
ML Systems Engineer, Large-Scale Model Training & RL Infrastructure
Palo Alto, California, United States
$195k-$262k/yr OnsiteFull Time
Strong Python and PyTorch skills, hands-on distributed model training and GPU cluster experience, debugging across NCCL/CUDA/PyTorch/Ray, and quantitative reasoning about throughput, utilization, memory, and cost.
Python, PyTorch, Megatron-LM, DeepSpeed, PyTorch FSDP/DTensor, Ray, verl, slime, AReaL, OpenRLHF, NCCL, CUDA, Triton, Nsight, InfiniBand, RDMA, RoCE, Slurm, Kubernetes
2mo
Save
Mark Applied
Hide
Research Scientist / Engineer - Efficient Modeling
Palo Alto, California, United States
OnsiteFull Time
Experience with model compression, quantization, pruning; PyTorch; transformers; hardware-aware optimization.
PyTorch, CUDA, TensorRT, Transformers
2mo
Save
Mark Applied
Hide
Machine Learning Engineer Intern - Planning
Santa Clara, California, United States
$19-$65/hr OnsiteInternship
0+ YOEInternship in ML/AI; experience with PyTorch/JAX; knowledge of trajectory prediction; strong communication.
PyTorch, JAX, PyTorch Lightning
2mo
Save
Mark Applied
Hide
ML Engineer, End-to-End Autonomy
Santa Clara, California, United States
$135k-$378k/yr HybridFull Time
Proven ML model development, production deployment for robotics, Python and PyTorch expertise, and cross-functional collaboration.
Python, PyTorch, ROS, CUDA, GPU
3mo
Save
Mark Applied
Hide
Staff Software Engineer, ML Infrastructure
San Francisco, California, United States
$220k-$260k/yr HybridFull Time
7+ YOE7+ years software systems, ML infra experience, Python, PyTorch, ML tooling, strong communication.
Python, PyTorch, AWS, TensorRT, ONNX, Weights & Biases, MLflow, ClearML
2mo
Save
Mark Applied
Hide
Staff Research Scientist, Perception Research
Mountain View or Kirkland or New York City
$251k-$310k/yr HybridFull Time
5+ YOEPhD preferred; 5+ years deep learning; experience with JAX, TensorFlow, PyTorch; strong Python skills.
Python, JAX, TensorFlow, PyTorch
3mo
Save
Mark Applied
Hide
Research Scientist Intern, AI Alignment
Bellevue or Menlo Park or Seattle or Boston or New York City or San Francisco
$8k-$12k/mo HybridInternship
Ph.D. in CS/ML/AI; work authorization; programming in Python/C/C++/Java; experience with PyTorch or TensorFlow.
Python, C++, C, Java, PyTorch, TensorFlow
3mo
Save
Mark Applied
Hide
Machine Learning Scientist (L4/L5) - Multi-modal Algorithms for Games
Los Gatos or Los Angeles
$466k-$750k/yr OnsiteFull Time
Expert in deep learning, transformers, diffusion; experience with LLMs/VLMs, multi-modal systems, optimization, and Python/PyTorch.
Python, PyTorch, Transformers, Diffusion, Reinforcement Learning, Machine Learning, Software Optimization
2mo
Save
Mark Applied
Hide
ML Inference Engineer
San Francisco, California, United States
OnsiteFull Time
Strong expertise in ML engineering, PyTorch, CUDA, and high-performance inference; experience with diffusion models and low-latency systems.
PyTorch, TensorRT, TransformerEngine, Nsight, ONNX Runtime, CUDA
1w
Save
Mark Applied
Hide
Founding Machine Learning - Eval Layer
San Francisco, California, United States
$150k-$275k/yr OnsiteFull Time
Strong Python and PyTorch coding; experience training VLMs or LLMs; and a track record developing and shipping evaluations for VLMs or LLMs.
Python, PyTorch, VLMs, LLMs
3mo
Save
Mark Applied
Hide
Multimodal LLMs Research Engineer
Sunnyvale or United States
OnsiteFull Time
2+ YOEPhD or MS with 2+ years industry experience; publications/patents; strong Python, PyTorch/JAX; AI tooling experience.
Python, PyTorch, JAX
1mo
Save
Mark Applied
Hide
Research Engineer/Scientist (all levels), Efficient Models
San Jose, California, United States
OnsiteFull Time
B.S. in CS or related, expertise in efficient generative/multimodal models, proficiency with PyTorch and JAX, strong communication and collaboration skills.
PyTorch, JAX
3mo
Save
Mark Applied
Hide
Research Engineer, Training & Inference
Palo Alto, California, United States
$200k-$450k/yr OnsiteFull Time
2+ YOEBS in CS or related field, 2+ years experience, Python proficiency, ML frameworks experience (PyTorch/JAX/TensorFlow).
Python, PyTorch, JAX, TensorFlow, C++, CUDA, NCCL, Triton, CuTeDSL, TileLang, CUDA, CUTLASS, ThunderKittens
2mo
Save
Mark Applied
Hide
Research Scientist - Vision Language Model
Sunnyvale, California, United States
$150k-$450k/yr OnsiteFull Time
Experience with large language models and vision-language models; strong Python and PyTorch; distributed training; multimodal data processing.
Python, PyTorch, Megatron, Triton, CUDA, Distributed training
2mo
Save
Mark Applied
Hide
Member of Technical Staff - Research
San Francisco, California, United States
OnsiteFull Time
Experience training or serving LLMs; RL environments or evaluation suites; PyTorch/JAX; experimental design; Master’s or research experience.
PyTorch, JAX
2mo
Save
Mark Applied
Hide
Senior ML/RL Engineer, Behavior Planning
Houston or San Francisco Bay Area
OnsiteFull Time
Deep RL experience in real-world systems; Python and PyTorch; MS or PhD in CS/Robotics; MARL and safety constraints knowledge.
Python, PyTorch
Explore Jobs
Expand Your Job Search
Top Cities in California for Pytorch
- Pytorch jobs in Los Angeles, CA
- Pytorch jobs in San Diego, CA
- Pytorch jobs in San Jose, CA
- Pytorch jobs in San Francisco, CA
- Pytorch jobs in Fresno, CA
- Pytorch jobs in Sacramento, CA
- Pytorch jobs in Long Beach, CA
- Pytorch jobs in Oakland, CA
- Pytorch jobs in Bakersfield, CA
- Pytorch jobs in Anaheim, CA
- Pytorch jobs in Santa Ana, CA
- Pytorch jobs in Riverside, CA
- Pytorch jobs in Stockton, CA
- Pytorch jobs in Chula Vista, CA
- Pytorch jobs in Irvine, CA
- Pytorch jobs in Fremont, CA
- Pytorch jobs in San Bernardino, CA
- Pytorch jobs in Fontana, CA
- Pytorch jobs in Modesto, CA
- Pytorch jobs in Oxnard, CA