4,804 pytorch jobs at 1,529 companies in United States

PromotedHiringCafe
Founding Machine Learning / AI Search Engineer
Cupertino, CA, US
$160k-$310k/yr On-SiteFull Time
HiringCafe
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.
Python, PyTorch, Elasticsearch, LLMs
3mo
Save
Mark Applied
Hide
Senior Life Cycle Engineer, PyTorch - Deep Learning
Santa Clara or United States
$152k-$288k/yr HybridFull Time
NVIDIA
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.
C++, Python, Scripting, CI/CD, Linux, Cloud
1mo
Save
Mark Applied
Hide
Compiler Engineer (PyTorch / Triton / LLVM / MLIR)
Los Altos, California, United States
OnsiteFull Time
Majestic Labs
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.
PyTorch, PyTorch Inductor, Dynamo, Triton, LLVM, MLIR, C/C++
1mo
Save
Mark Applied
Hide
Staff Compiler Engineer - PyTorch + Kernel DSLPLATE
San Jose, California, United States
$163k-$253k/yr OnsiteFull Time
Samsung Semiconductor
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.
Triton, Helion, MLIR, XLA, TVM, Inductor, IREE, CUTLASS, PyTorch, LLVM/MLIR
3mo
Save
Mark Applied
Hide
Senior Life Cycle Engineer, PyTorch - Deep Learning
Santa Clara or United States
$152k-$288k/yr HybridFull Time
NVIDIA
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.
C++, Python, Linux, CI/CD
2d
Save
Mark Applied
Hide
ML Systems Engineer, Large-Scale Model Training & RL Infrastructure
Palo Alto, California, United States
$195k-$262k/yr OnsiteFull Time
Nebius
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.
Python, PyTorch, Megatron-LM, DeepSpeed, PyTorch FSDP/DTensor, Ray, verl, slime, AReaL, OpenRLHF, NCCL, CUDA, Triton, Nsight, InfiniBand, RDMA, RoCE, Slurm, Kubernetes
1mo
Save
Mark Applied
Hide
Machine Learning Engineer (m/f/d)
Ljubljana or United States or Europe or Asia
OnsiteFull Time
Sportradar
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.
Python, PyTorch, PyTorch Lightning, AWS, Docker, Singularity, JIRA
2mo
Save
Mark Applied
Hide
Research Scientist / Engineer - Efficient Modeling
Palo Alto, California, United States
OnsiteFull Time
Rhoda AI
Rhoda AI: Developing generalist robotic intelligence for real-world industrial automation.
Experience with model compression, quantization, pruning; PyTorch; transformers; hardware-aware optimization.
PyTorch, CUDA, TensorRT, Transformers
2mo
Save
Mark Applied
Hide
Senior Research Scientist (PREP0004541)
Gaithersburg, Maryland, United States
$62k-$84k/yr OnsiteFull Time
Johns Hopkins University
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.
PyTorch, TensorFlow, PyTorch Geometric, RDKit, pysmiles
2mo
Save
Mark Applied
Hide
Machine Learning Engineer Intern - Planning
Santa Clara, California, United States
$19-$65/hr OnsiteInternship
PlusAI
PlusAI: AI-based virtual driver software for factory-built autonomous trucks.
0+ YOEInternship in ML/AI; experience with PyTorch/JAX; knowledge of trajectory prediction; strong communication.
PyTorch, JAX, PyTorch Lightning
2mo
Save
Mark Applied
Hide
Data Scientist
Lackland AFB, Texas, United States
$94k-$197k/yr OnsiteFull Time
CACI
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.
PyTorch, TensorFlow, Spark MLlib, Scikit-learn, XGBoost, Keras, Hugging Face, PyTorch Geometric, MLflow, SHAP, GNNExplainer, Attention mechanisms
3mo
Save
Mark Applied
Hide
Machine Learning Engineer, Discovery Recommendations
Cary or United States
$148k-$271k/yr HybridFull Time
Epic Games
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.
Python, PyTorch, TorchRec, Transformers, Ray, Databricks, AWS, Spark
3mo
Save
Mark Applied
Hide
Research Scientist - LLM
Redwood City, California, United States
$225k-$400k/yr OnsiteFull Time
Retell AI
Retell AI: Develops human-like AI voice agents for automated call centers.
ML researcher with strong PyTorch experience; work on LLMs, audio models, and real-time systems.
PyTorch, LLMs, Audio models
3mo
Save
Mark Applied
Hide
Machine Learning Engineer, Ads Personalization (New York, NY, US, 10036)
New York, New York, United States
$124k-$150k/yr OnsiteFull Time
Paramount
ParamountNASDAQ: PSKY: Produces and distributes media content across global entertainment platforms.
3+ YOE3+ years in machine learning; Python/SQL; TensorFlow or PyTorch; Google Cloud Platform experience.
Python, SQL, TensorFlow, PyTorch, GCP, Qdrant
2mo
Save
Mark Applied
Hide
Senior Machine Learning Engineer – Physical AI
Wilmington, Massachusetts, United States
$140k-$165k/yr OnsiteFull Time
Goddard
Goddard: Engineering and industrial design firm for medical and robotic products
5+ YOE5+ years in ML engineering with production shipping, Python with PyTorch/TensorFlow, edge deployment experience.
Python, PyTorch, TensorFlow, TFLite, ONNX, TensorRT, CoreML, MLflow, Weights & Biases, CI/CD
3mo
Save
Mark Applied
Hide
Machine Learning Engineer - Visual Agents - Special Projects
Cupertino, California, United States
OnsiteFull Time
Apple
AppleNASDAQ: AAPL: Designs and sells consumer electronics, software, and online services.
2+ YOE2+ years in building/evaluating multimodal or vision-language models; Python and ML frameworks (PyTorch or TensorFlow).
Python, PyTorch, TensorFlow, Vision-Language Models
2mo
Save
Mark Applied
Hide
AI Inference Engineer Intern - Model Pruning
Burlingame, California, United States
$45-$60/hr OnsiteTemporary
Quadric
Quadric: Designing licensable processor IP for on-device AI inference.
MS student in CS; proficient in Python and PyTorch; experience with model pruning, quantization, and vision model metrics.
Python, PyTorch
3mo
Save
Mark Applied
Hide
Machine Learning Internship - PhD: 2027
Bala Cynwyd, Pennsylvania, United States
OnsiteInternship
Susquehanna International Group
Susquehanna International Group: Global quantitative trading and market making firm.
PhD candidate with ML/data science experience; strong publication record; hands-on ML with PyTorch/TensorFlow.
PyTorch, TensorFlow
3mo
Save
Mark Applied
Hide
Research Engineer, World Models
Toronto or San Francisco or Pittsburgh
$155k-$269k/yr HybridFull Time
Waabi
Waabi: Develops Physical AI for autonomous trucking and robotaxis.
Strong Python & PyTorch (or JAX) skills; distributed training; large-scale model deployment; team collaboration.
Python, PyTorch, JAX, Distributed training, Kubeflow, Ray, ONNX, TensorRT
3mo
Save
Mark Applied
Hide
Sr. Computer Vision Engineer (Deep Learning)
Mountain View or Garden Grove
$180k-$240k/yr OnsiteFull Time
Harbinger
Harbinger: Produces electric chassis platforms for medium-duty commercial transport vehicles.
5+ YOE5+ years in DL perception models for ADAS; Python proficiency; PyTorch or TensorFlow; embedded deployment; collaboration.
Python, PyTorch, TensorFlow, Linux
1mo
Save
Mark Applied
Hide
Machine Learning Scientist, Multimodal AI
United States
$125k-$156k/yr RemoteFull Time
Natera
NateraNASDAQ: NTRA: Provides clinical genetic testing and diagnostics services.
PhD in a quantitative field; biomedical ML experience; Python and PyTorch; AWS; experience with multimodal biomedical data.
PyTorch, Python, AWS, Deep Learning, Transformers