$122K - $161K/yr
PyTorch, JAX, TensorFlow, ONNX, etc) and ideally inference engines and runtimes such as vLLM, SGLang, and MLC. * Strong Python and C/C++ programming skills * Strong experience in GPU kernel ...
$122K - $161K/yr
PyTorch, JAX, TensorFlow, ONNX, etc) and ideally inference engines and runtimes such as vLLM, SGLang, and MLC. * Strong Python and C/C++ programming skills * Strong experience in GPU kernel ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Strong programming skills in Python, with experience in frameworks like TensorFlow, PyTorch, or Hugging Face. * Proficiency in designing and deploying machine learning models, particularly in ...
Hillsboro, OR · On-site
$170K - $315K/yr
Implementation of linear algebra algorithms (functions from BLAS, LAPACK, or PyTorch) * Performance engineering and software performance optimizations * Floating point arithmetic and numerical ...
Hillsboro, OR · On-site
$170K - $315K/yr
Implementation of linear algebra algorithms (functions from BLAS, LAPACK, or PyTorch) * Performance engineering and software performance optimizations * Floating point arithmetic and numerical ...
Hillsboro, OR · On-site
$170K - $315K/yr
Implementation of linear algebra algorithms (functions from BLAS, LAPACK, or PyTorch) * Performance engineering and software performance optimizations * Floating point arithmetic and numerical ...
Hillsboro, OR · On-site
$170K - $315K/yr
Implementation of linear algebra algorithms (functions from BLAS, LAPACK, or PyTorch) * Performance engineering and software performance optimizations * Floating point arithmetic and numerical ...
OR · Remote
Collaborate with cross-functional teams including DevOps, cybersecurity, QA, and product ... Experience with AI/ML technologies, including machine learning frameworks (TensorFlow, PyTorch) or ...
OR · Remote
Collaborate with cross-functional teams including DevOps, cybersecurity, QA, and product ... Experience with AI/ML technologies, including machine learning frameworks (TensorFlow, PyTorch) or ...
OR · On-site
$25/hr
Develop SageMaker training jobs for NLP models (NeMo, PyTorch) * Implement MLflow tracking and model registry integrations * Write infrastructure-as-code using Terraform (AWS S3, IAM, VPC) * Create ...
OR · On-site +1
$140K - $190K/yr
Proficiency in Python and its ML ecosystem (pandas, scikit-learn, TensorFlow/PyTorch), with clean ... Experience with functional programming (e.g. Clojure) is a plus but not required. * Cloud and MLOps ...
OR · On-site +1
$140K - $190K/yr
Proficiency in Python and its ML ecosystem (pandas, scikit-learn, TensorFlow/PyTorch), with clean ... Experience with functional programming (e.g. Clojure) is a plus but not required. * Cloud and MLOps ...
The ML Ops Engineer will work at the intersection of advanced AI/ML development, machine learning ... Learn, MLlib, TensorFlow, PyTorch, etc. * Practical experience in deploying AI/ML models in ...
The ML Ops Engineer will work at the intersection of advanced AI/ML development, machine learning ... Learn, MLlib, TensorFlow, PyTorch, etc. * Practical experience in deploying AI/ML models in ...
Corvallis, OR · Remote
$72K - $97K/yr
More typical DevOps responsibilities for software development as required. Requirements Required ... Experience with Python and ML frameworks (PyTorch, TensorFlow, or similar) * Experience building CI ...
Corvallis, OR · Remote
$72K - $97K/yr
More typical DevOps responsibilities for software development as required. Requirements Required ... Experience with Python and ML frameworks (PyTorch, TensorFlow, or similar) * Experience building CI ...
Corvallis, OR · Remote
$72K - $97K/yr
More typical DevOps responsibilities for software development as required. Required Qualifications ... Experience with Python and ML frameworks (PyTorch, TensorFlow, or similar) * Experience building CI ...
Corvallis, OR · Remote
$72K - $97K/yr
More typical DevOps responsibilities for software development as required. Required Qualifications ... Experience with Python and ML frameworks (PyTorch, TensorFlow, or similar) * Experience building CI ...
Myrtle Point, OR · On-site +1
$165K - $190K/yr
Strong programming skills in Python; experience with ML frameworks (PyTorch, TensorFlow) and agent orchestration tools. * Experience in business process analysis, process mapping, and workflow ...
Myrtle Point, OR · On-site +1
$165K - $190K/yr
Strong programming skills in Python; experience with ML frameworks (PyTorch, TensorFlow) and agent orchestration tools. * Experience in business process analysis, process mapping, and workflow ...
We're hiring an Engineering Manager, Machine Learning and Computer Vision to lead a team of ... Hands-on expertise with computer vision, deep learning (e.g., PyTorch), model training/evaluation ...
We're hiring an Engineering Manager, Machine Learning and Computer Vision to lead a team of ... Hands-on expertise with computer vision, deep learning (e.g., PyTorch), model training/evaluation ...
$165K - $190K/yr
Strong programming skills in Python; experience with ML frameworks (PyTorch, TensorFlow) and agent orchestration tools. * Experience in business process analysis, process mapping, and workflow ...
$165K - $190K/yr
Strong programming skills in Python; experience with ML frameworks (PyTorch, TensorFlow) and agent orchestration tools. * Experience in business process analysis, process mapping, and workflow ...
$104K - $143K/yr
... g., PyTorch, TensorFlow) * Hands-on experience with MLOps frameworks and workflow tooling (e.g ... Understanding of CI/CD workflows and DevOps practices applied to ML systems (e.g., Git, Code Review ...
New
$122K - $232K/yr
Experience with AI/ML frameworks and libraries (e.g., Pytorch, CUDA, vLLM, Triton, NCCL, oneCCL ... programmable solutions, and the promise of always-on 5G connectivity, Intel is disrupting ...
$122K - $232K/yr
Experience with AI/ML frameworks and libraries (e.g., Pytorch, CUDA, vLLM, Triton, NCCL, oneCCL ... programmable solutions, and the promise of always-on 5G connectivity, Intel is disrupting ...
OR · On-site +1
We are looking for Staff / Sr Staff Software Engineers who are passionate about helping to deliver ... Experience with common machine learning algorithms and libraries (xgboost, sklearn, pytorch, pandas ...
OR · On-site +1
We are looking for Staff / Sr Staff Software Engineers who are passionate about helping to deliver ... Experience with common machine learning algorithms and libraries (xgboost, sklearn, pytorch, pandas ...
OR · Remote
$122K - $161K/yr
Develop internal developer tools and CLIs (often in Go) that improve how we build, test, and ship ... PyTorch, TensorFlow) and deploying them in cloud/Kubernetes environments. * Prior experience in SRE ...
OR · Remote
$122K - $161K/yr
Develop internal developer tools and CLIs (often in Go) that improve how we build, test, and ship ... PyTorch, TensorFlow) and deploying them in cloud/Kubernetes environments. * Prior experience in SRE ...
| Aspect | Pytorch Developer | Machine Learning Engineer |
|---|---|---|
| Required Credentials | Bachelor's or higher in CS, experience with PyTorch | Bachelor's or higher in CS, data science, or related field, with ML experience |
| Work Environment | Research labs, AI startups, tech companies focusing on deep learning | Tech companies, finance, healthcare, often involving deployment and scaling ML models |
| Industry Usage | Primarily in AI research and development teams | Across industries implementing ML solutions in production |
While both roles require knowledge of machine learning and experience with PyTorch, a Pytorch Developer mainly focuses on developing and optimizing deep learning models using PyTorch. A Machine Learning Engineer often has a broader scope, including deploying, maintaining, and scaling ML models across various platforms and industries.
$122K - $161K/yr
Full-time
Posted 8 days ago
We're looking for outstanding AI systems engineers to develop groundbreaking technologies in the inference systems software stack! We build innovative AI systems software to accelerate for AI inference. As a member of the team, you'll develop libraries, code generators, and GPU kernel technologies for NVIDIA's hardware architecture. This means designing and building things like new abstractions, efficient attention kernel implementations, new LLM inference runtimes components, and kernel code generators to accelerate large language models, agents, and other high-impact AI workloads.
What you'll be doing:
Innovating and developing new AI systems technologies for efficient inference
Designing, implementing, and optimizing kernels for high impact AI workloads
Designing and implementing extensible abstractions for LLM serving engines
Building efficient just-in-time domain specific compilers and runtimes
Collaborating closely with other engineers at NVIDIA across deep learning frameworks, libraries, kernels, and GPU arch teams
Contributing to open source communities like FlashInfer, vLLM, and SGLang
What we need to see:
Masters degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); PhD are preferred
6+ years (academic/ industry) experience with ML/DL systems development preferable
Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc) and ideally inference engines and runtimes such as vLLM, SGLang, and MLC.
Strong Python and C/C++ programming skills
Strong experience in GPU kernel development and performance optimizations (especially using CUDA C/C++, cuTile, Triton, or similar)
Ways to stand out from the crowd:
Background in domain specific compiler and library solutions for LLM inference and training (e.g. FlashInfer, Flash Attention)
Expertise in inference engines like vLLM and SGLang
Expertise in machine learning compilers (e.g. Apache TVM, MLIR)
Open source project ownership or contributions
You will also be eligible for equity and benefits.
This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes.
NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.Sourced by ZipRecruiter
NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology--and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent.
Computer and electronic product manufacturing
10,000+ Employees
Santa Clara, CA, US
1993