PyTorch with Triton performance Engineer Bellevue, WA (Onsite) Contract / FTE Job Summary Design and implement high intensity stress workloads using PyTorch and Triton to identify performance ...
PyTorch with Triton performance Engineer Bellevue, WA (Onsite) Contract / FTE Job Summary Design and implement high intensity stress workloads using PyTorch and Triton to identify performance ...
Python Automation Developer (Onsite)
Seattle, WA · On-site
$57.25 - $78.75/hr
Hi, Job Title: - Python Automation Developer Location: - Renton, WA / Everett, WA Full-time ... TensorFlow and PyTorch; 4. Framework: Django ; Flask; FastAPI 5. Web Scraping and HTTP: Beautiful ...
Python Automation Developer (Onsite)
Seattle, WA · On-site
$57.25 - $78.75/hr
Hi, Job Title: - Python Automation Developer Location: - Renton, WA / Everett, WA Full-time ... TensorFlow and PyTorch; 4. Framework: Django ; Flask; FastAPI 5. Web Scraping and HTTP: Beautiful ...
Senior Software Engineer, AI Networking
Seattle, WA · On-site
$139K - $183K/yr
Knowledge in PyTorch, CUDA, and NCCL libraries. * Proven software engineering/development skills ... With competitive salaries and a comprehensive benefits package, NVIDIA is widely regarded as one of ...
Senior Software Engineer, AI Networking
Seattle, WA · On-site
$139K - $183K/yr
Knowledge in PyTorch, CUDA, and NCCL libraries. * Proven software engineering/development skills ... With competitive salaries and a comprehensive benefits package, NVIDIA is widely regarded as one of ...
Engineering Manager, AI Developer Tools
$188K - $325K/yr
The Developer Experience team delivers the essential tools and systems that empower hundreds of ... Familiarity with AI/ML frameworks (e.g., PyTorch, TensorFlow) and LLM libraries (e.g., LangChain ...
Engineering Manager, AI Developer Tools
$188K - $325K/yr
The Developer Experience team delivers the essential tools and systems that empower hundreds of ... Familiarity with AI/ML frameworks (e.g., PyTorch, TensorFlow) and LLM libraries (e.g., LangChain ...
Software Development Engineer - AI/ML, Amazon Neuron, Multimodal Inference
Seattle, WA · On-site
$159K/yr
Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels for ML functions ...
Software Development Engineer - AI/ML, Amazon Neuron, Multimodal Inference
Seattle, WA · On-site
$159K/yr
Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels for ML functions ...
Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels for ML functions ...
Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels for ML functions ...
Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels for ML functions ...
Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic infrastructure, innovate new methods and create high-performance kernels for ML functions ...
Senior Deep Learning Software Engineer
Redmond, WA · Hybrid
$137K - $180K/yr
We are looking for a Senior Deep Learning Software Engineer to design and build our automated ... Contributions to PyTorch, JAX, or other Machine Learning Frameworks. * Knowledge of GPU ...
Senior Deep Learning Software Engineer
Redmond, WA · Hybrid
$137K - $180K/yr
We are looking for a Senior Deep Learning Software Engineer to design and build our automated ... Contributions to PyTorch, JAX, or other Machine Learning Frameworks. * Knowledge of GPU ...
ML Model Serving Engineer
Bellevue, WA · On-site
Required : • Expert in some differentiable array computing framework, preferably PyTorch. • ... Significant systems programming experience; ex. Experience working on high-performance server ...
ML Model Serving Engineer
Bellevue, WA · On-site
Required : • Expert in some differentiable array computing framework, preferably PyTorch. • ... Significant systems programming experience; ex. Experience working on high-performance server ...
Staff AI Software Engineer, Edge Model Optimization & Deployment
Seattle, WA · On-site
$70K - $300K/yr
We are seeking an accomplished Staff AI Software Engineer - Edge Model Optimization & Deployment to ... Convert and optimize 2D/3D CNNs and Transformer-based models (PyTorch/TensorFlow - ONNX - TensorRT ...
Staff AI Software Engineer, Edge Model Optimization & Deployment
Seattle, WA · On-site
$70K - $300K/yr
We are seeking an accomplished Staff AI Software Engineer - Edge Model Optimization & Deployment to ... Convert and optimize 2D/3D CNNs and Transformer-based models (PyTorch/TensorFlow - ONNX - TensorRT ...
Orchestrate distributed training runs across 1,000+ GPUs using PyTorch, DeepSpeed, or Megatron-LM. * Optimize networking (InfiniBand/RDMA) and memory management to prevent out-of-memory errors.
Orchestrate distributed training runs across 1,000+ GPUs using PyTorch, DeepSpeed, or Megatron-LM. * Optimize networking (InfiniBand/RDMA) and memory management to prevent out-of-memory errors.
Staff AI Software Engineer, Edge Model Optimization & Deployment
Seattle, WA · On-site
$70K - $300K/yr
We are seeking an accomplished Staff AI Software Engineer - Edge Model Optimization & Deployment to ... Convert and optimize 2D/3D CNNs and Transformer-based models (PyTorch/TensorFlow ONNX TensorRT ...
Staff AI Software Engineer, Edge Model Optimization & Deployment
Seattle, WA · On-site
$70K - $300K/yr
We are seeking an accomplished Staff AI Software Engineer - Edge Model Optimization & Deployment to ... Convert and optimize 2D/3D CNNs and Transformer-based models (PyTorch/TensorFlow ONNX TensorRT ...
We are seeking an accomplished Staff AI Software Engineer - Edge Model Optimization & Deployment to ... Convert and optimize 2D/3D CNNs and Transformer-based models (PyTorch/TensorFlow → ONNX → ...
Quick apply
We are seeking an accomplished Staff AI Software Engineer - Edge Model Optimization & Deployment to ... Convert and optimize 2D/3D CNNs and Transformer-based models (PyTorch/TensorFlow → ONNX → ...
LLM Pre-training & Distributed Engineer (AI Infrastructure)
Seattle, WA · On-site
$122K - $160K/yr
Orchestrate distributed training runs across 1,000+ GPUs using PyTorch, DeepSpeed, or Megatron-LM. * Optimize networking (InfiniBand/RDMA) and memory management to prevent out-of-memory errors.
LLM Pre-training & Distributed Engineer (AI Infrastructure)
Seattle, WA · On-site
$122K - $160K/yr
Orchestrate distributed training runs across 1,000+ GPUs using PyTorch, DeepSpeed, or Megatron-LM. * Optimize networking (InfiniBand/RDMA) and memory management to prevent out-of-memory errors.
Senior Software Engineer - AI Frameworks
Redmond, WA · On-site
$160K - $261K/yr
We build software across the stack, from PyTorch and inference systems such as vLLM and SGLang to ... We are seeking a self-motivated Senior Software Engineer - AI Frameworks who thrives on technical ...
Senior Software Engineer - AI Frameworks
Redmond, WA · On-site
$160K - $261K/yr
We build software across the stack, from PyTorch and inference systems such as vLLM and SGLang to ... We are seeking a self-motivated Senior Software Engineer - AI Frameworks who thrives on technical ...
Python Engineer
Redmond, WA · On-site
$75 - $80/hr
We are currently seeking a Python Engineer for our client in the IT Services domain. We value our ... PyTorch (even though the role is not ML-heavy). Team Culture: - Strong emphasis on collaboration ...
Quick apply
Python Engineer
Redmond, WA · On-site
$75 - $80/hr
We are currently seeking a Python Engineer for our client in the IT Services domain. We value our ... PyTorch (even though the role is not ML-heavy). Team Culture: - Strong emphasis on collaboration ...
Senior Machine Learning Compiler Engineer
$139K - $183K/yr
... such as PyTorch, TensorFlow and MxNet. Amazon Neuron and Inferentia are used at scale with ... The team covers multiple disciplines including silicon engineering, hardware design and ...
Senior Machine Learning Compiler Engineer
$139K - $183K/yr
... such as PyTorch, TensorFlow and MxNet. Amazon Neuron and Inferentia are used at scale with ... The team covers multiple disciplines including silicon engineering, hardware design and ...
Senior Machine Learning Compiler Engineer
Seattle, WA · On-site
$139K - $183K/yr
... such as PyTorch, TensorFlow and MxNet. Amazon Neuron and Inferentia are used at scale with ... The team covers multiple disciplines including silicon engineering, hardware design and ...
Senior Machine Learning Compiler Engineer
Seattle, WA · On-site
$139K - $183K/yr
... such as PyTorch, TensorFlow and MxNet. Amazon Neuron and Inferentia are used at scale with ... The team covers multiple disciplines including silicon engineering, hardware design and ...
Systems Engineer
Redmond, WA · On-site
$155K - $205K/yr
Optimize GPU/CUDA workloads and accelerate ML frameworks (PyTorch, JAX, TensorFlow) for robotics ... Strong systems programming skills in one or more of: C++, Rust, Go, Python. * Solid understanding ...
Systems Engineer
Redmond, WA · On-site
$155K - $205K/yr
Optimize GPU/CUDA workloads and accelerate ML frameworks (PyTorch, JAX, TensorFlow) for robotics ... Strong systems programming skills in one or more of: C++, Rust, Go, Python. * Solid understanding ...
... PyTorch, Ray) • Work with workflow orchestration tools (e.g., Airflow, Flyte, or similar) to ... engineers to support experimentation and model iteration • Help optimize performance and ...
... PyTorch, Ray) • Work with workflow orchestration tools (e.g., Airflow, Flyte, or similar) to ... engineers to support experimentation and model iteration • Help optimize performance and ...
Pytorch Developer information
What is a PyTorch Developer?
What are the key skills and qualifications needed to thrive as a Pytorch Developer, and why are they important?
What is the difference between Pytorch Developer vs Machine Learning Engineer?
| 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.
What are some common challenges Pytorch Developers face when deploying machine learning models to production environments?
- Software Engineer Internship
- Summer Software Engineer Internship
- Internship 3D Modeling
- Entry Level Software Engineer Internship
- Optics Internship
- Linux System Engineer Internship
- Network Security Engineer Internship
- Internship Housing Relocation Welding
- Internship Ruby Developer
- Simulation Software Engineer Internship
- Industrial Engineering Internship
- Internship Airbnb Software Engineer
- Internship Ngo Software Engineer
- Internship Palantir Software Engineer
- Internship Government Software Engineering
- Neural Engineering Internship
- Internship Deloitte Mechanical Engineering
- Online Software Engineering Internships
- Internship Drone Software Engineer
- Internship Traffic Monitoring

Other
Posted 2 days ago
Job description
PyTorch with Triton performance Engineer
Bellevue, WA (Onsite)
Contract / FTE
Job Summary
Design and implement high intensity stress workloads using PyTorch and Triton to identify performance bottlenecks and improve platform stability and maturity
Job Description
Design and implement high intensity stress workloads using PyTorch and Triton Exercise core MAIA execution paths including compute memory DMA and collectives Enable early detection of performance cliffs stability issues and system bottlenecks across simulator and real hardware Improve platform maturity reduce latestage escapes and increase confidence for broader internal and external adoption Develop PyTorch workloads stressing modellevel execution such as large GEMMs attention patterns MoElike behavior mixed precision and longrunning loops Author custom Triton kernels to stress hardware execution units memory hierarchies and synchronization paths Build parameterized stress harnesses scalable by problem size number of devices and runtime duration Integrate workloads with existing profiling monitoring and failure triage tooling Collaborate with platform firmware and SDK teams to target known risk areas and emerging issues Document usage patterns and provide reproducible scripts for lab and continuous integration CI usage
Roles and Responsibilities : Develop and maintain a library of reusable PyTorch stress workloads Create Tritonbased micro and macrokernels designed specifically for stress and saturation testing Build and support test harnesses and scripts for singledevice and multidevice execution Ensure workload designs align with platform risk areas and emerging hardwaresoftware issues Collaborate crossfunctionally with platform firmware and SDK teams to refine stress tests Provide comprehensive documentation describing workload intent configuration options and expected stress characteristics Support profiling monitoring and failure triage by integrating stress workloads with existing tools Deliver reproducible and scalable testing solutions for lab and CI environments
About V Dart
Sourced by ZipRecruiter
Company size
1,001 - 5,000 Employees
Headquarters location
Alpharetta, GA, US
Year founded
2008