4,804 pytorch jobs at 1,529 companies in United States
🚀PromotedHiringCafe
Founding Machine Learning / AI Search Engineer
Cupertino, CA, US
$160k-$310k/yrOn-SiteFull Time
HiringCafe: Building a 100× better job search engine to take on Indeed and LinkedIn.
Build the ML and AI search behind HiringCafe — ranking, recommenders, retrieval, and LLM agents that surface jobs people would never find on their own.
Senior Life Cycle Engineer, PyTorch - Deep Learning
Santa Clara or United States
$152k-$288k/yrHybridFull Time
NVIDIANASDAQ: NVDA: Designs GPU-accelerated computing and artificial intelligence hardware.
5+ YOEMS or PhD in Computer Science or Engineering; 5+ years in large software projects; proficiency in C++ and/or Python; scripting; Linux CI/CD; cloud-hosted developer environments.
Majestic Labs: Developing memory-first AI server platforms for data centers.
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.
Samsung SemiconductorKorea Exchange: 005930: Designs and manufactures memory chips, processors, and sensors.
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.
Senior Life Cycle Engineer, PyTorch - Deep Learning
Santa Clara or United States
$152k-$288k/yrHybridFull Time
NVIDIANASDAQ: NVDA: Designs graphics processing units and artificial intelligence hardware.
5+ YOEMS/PhD or equivalent experience, 5+ years building/supporting large software projects, proficiency in C++ and/or Python, scripting, continuous build/release for Linux, and cloud CI/CD experience.
ML Systems Engineer, Large-Scale Model Training & RL Infrastructure
Palo Alto, California, United States
$195k-$262k/yrOnsiteFull Time
NebiusNasdaq: NBIS: Builds cloud infrastructure and software for artificial intelligence development.
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.
SportradarNASDAQ: SRAD: Provides data and technology for sports, media, and betting.
Expertise in modern ML and deep learning for sequence modeling and generative AI; strong Python and PyTorch/PyTorch Lightning skills; experience with distributed training, AWS, container-based development, data pipelines, and production deployment.
Johns Hopkins University: Private research university providing higher education and medical services.
5+ YOEPh.D. in a related field, 5+ years ML experience in biochemistry/drug design, expertise in generative and predictive AI, NLP and GNNs, proficiency with PyTorch, TensorFlow, PyTorch Geometric, RDKit, pysmiles, strong communication and manuscript writing skills.
CACINYSE: CACI: Provides information technology and professional services to government clients.
5+ YOEClear TS/SCI clearance; 5+ years ML in production; Master’s in DS/AI; PyTorch/TensorFlow; strong Python; ML libraries (Spark MLlib, Scikit-learn, XGBoost, Keras, Hugging Face, PyTorch Geometric, MLflow) and explainability tools.
Epic Games: Develops video games and licenses the Unreal game engine.
3+ YOE3-5+ years in production recommendation/ranking systems; Python with PyTorch; experience with large-scale data processing and cloud ML; knowledge of NLP/vision or generative AI as ranking features; experience with PyTorch TorchRec, Transformers, Ray, Databricks, AWS.