1

Pytorch Developer Jobs in Las Vegas, NV (NOW HIRING)

Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading ... Data Pipeline Engineering : Optimize robust data loading pipelines that maximize training ...

Software Engineer, ML Dev Enablement

Las Vegas, NV · On-site

$123K - $163K/yr

  • Medical

  • Dental

  • Vision

  • Life

  • Retirement

Experience with ML frameworks such as PyTorch or Ray. * Experience building, integrating, or enhancing Agentic AI systems and LLM-driven developer tools. We encourage a hybrid schedule with in-office ...

Software Engineer, ML Dev Enablement

Las Vegas, NV · On-site +1

$123K - $163K/yr

  • Medical

  • Dental

  • Vision

  • Life

  • Retirement

Experience with ML frameworks such as PyTorch or Ray. * Experience building, integrating, or enhancing Agentic AI systems and LLM-driven developer tools. We encourage a hybrid schedule with in-office ...

Experience with ML frameworks such as PyTorch or Ray. * Experience building, integrating, or enhancing Agentic AI systems and LLM-driven developer tools. We encourage a hybrid schedule with in-office ...

Lead Artificial Intelligence Engineer

Las Vegas, NV · On-site

$99K - $130K/yr

Python, PyTorch, TensorFlow, JAX * LangChain, semantic search, vector embeddings * Prompt engineering & LLM orchestration frameworks Preferred * Bachelor's degree in Computer Science, Engineering ...

Must-Have Skills 3+ years of ML engineering experience -- model training, fine-tuning, or post-training pipelines in research or production Strong Python and deep learning proficiency (PyTorch ...

Must-Have Skills 3+ years of ML engineering experience -- model training, fine-tuning, or post-training pipelines in research or production Strong Python and deep learning proficiency (PyTorch ...

Programming & Tooling * Python, PyTorch, TensorFlow, JAX * LangChain, semantic search, vector embeddings * Prompt engineering & LLM orchestration frameworks * Excellent communication, problem-solving ...

next page

Showing results 1-20

Pytorch Developer information

What is a PyTorch developer?

A PyTorch Developer is a software engineer or data scientist who specializes in using PyTorch, an open-source machine learning library, to build and deploy deep learning models. Their responsibilities typically include designing neural network architectures, training and evaluating models, and optimizing code for performance. PyTorch Developers work in fields such as artificial intelligence, computer vision, and natural language processing, collaborating with teams to solve complex problems using machine learning. They are proficient in Python and have a strong understanding of deep learning concepts. Additionally, they often contribute to research, development, and the deployment of AI solutions in production environments.

What are some common challenges PyTorch developers face when deploying machine learning models to production environments?

Pytorch Developers often encounter challenges when transitioning models from research to production, such as optimizing model performance for inference speed and memory usage, ensuring compatibility with deployment frameworks like TorchScript or ONNX, and managing dependencies across different systems. Additionally, integrating PyTorch models into existing software stacks and maintaining reproducibility can be complex. Collaborating closely with DevOps and data engineering teams is crucial to address these issues and ensure smooth deployment.

What are the key skills and qualifications needed to thrive as a PyTorch developer, and why are they important?

To thrive as a Pytorch Developer, you need strong programming skills in Python, a solid grasp of machine learning concepts, and experience with deep learning frameworks—especially PyTorch itself. Familiarity with tools like CUDA, Jupyter Notebooks, and version control systems (e.g., Git) is typically expected, along with knowledge of cloud platforms or relevant certifications. Problem-solving ability, effective collaboration, and clear communication are crucial soft skills for success in this role. These skills and qualities are vital for efficiently building, optimizing, and deploying machine learning models in real-world applications.

What is the difference between Pytorch Developer vs Machine Learning Engineer?

AspectPytorch DeveloperMachine Learning Engineer
Required CredentialsBachelor's or higher in CS, experience with PyTorchBachelor's or higher in CS, data science, or related field, with ML experience
Work EnvironmentResearch labs, AI startups, tech companies focusing on deep learningTech companies, finance, healthcare, often involving deployment and scaling ML models
Industry UsagePrimarily in AI research and development teamsAcross 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.

Infographic showing various Pytorch Developer job openings in Las Vegas, NV as of August 2026, with employment types broken down into 82% Full Time, 5% Part Time, and 13% Contract. Highlights an 83% Physical, 5% Hybrid, and 12% Remote job distribution.

Software Engineer, ML Dev Enablement

Motional

Las Vegas, NV • On-site, Remote

Full-time

Re-posted 20 days ago


Job description

Mission Summary:

We are looking for a Software Engineer to join our ML Infrastructure: Dev Enablement Team. Our mission is to build a frictionless development environment that empowers our researchers and engineers to rapidly innovate on deep learning models for autonomous driving.

We manage a high-scale Cloud Development Environment (CDE) platform that provides standardized, high-performance workspaces for ML development. As we evolve, in this role, you'll spearhead high-impact initiatives: designing multi-cloud setups to maximize GPU availability, driving deep-level model optimization, and building next-generation Agentic AI toolings. You will play a pivotal role in ensuring our training ecosystem remains cutting-edge, resilient and highly efficient.

What You'll Be Doing:

  • Build Agentic AI Tooling: Design, develop, and enhance Agentic AI tools and systems to automate workflows, streamline the ML lifecycle, and empower developer productivity.
  • Scale Core Infrastructure: Drive the continuous development of our core ML infrastructure and existing CDE platform, leveraging Kubernetes to build robust, high-scale distributed solutions.
  • System-Level ML Optimization: Partner closely with ML Researchers to profile and optimize distributed training jobs (PyTorch/DDP) and data pipelines. Focus on resolving system-level bottlenecks-such as data loading (I/O), memory management, and network communication overhead-to maximize GPU utilization and training throughput.
  • Collaborate Cross-Functionally: Partner with ML engineers and data scientists to understand their complex needs, bridging the gap between underlying infrastructure and model development.

What We're Looking For:

  • BS or MS in Computer Science or related field
  • Strong knowledge of software engineering principles and distributed systems.
  • Strong proficiency with Python or Go or C++
  • Experience with building on AWS services or other Cloud platforms and container orchestration using Kubernetes.
  • Experience with the various stages of the ML development lifecycle
Bonus Points:
  • Hands-on experience with ML model profiling and performance optimization for distributed training.
  • Experience managing or working with high-performance compute resources (GPUs).
  • Experience with ML frameworks such as PyTorch or Ray.
  • Experience building, integrating, or enhancing Agentic AI systems and LLM-driven developer tools.

 We encourage a hybrid schedule with in-office time at our Las Vegas location to support collaboration, or this role can be fully remote.