PyTorch Software Engineer Jobs

Develop perception capabilities that help autonomous air, maritime, and ground systems understand objects, motion, sensors, and terrain. This research-to-product role combines computer vision and machine learning experimentation with production C++ and Python development for edge deployment in the Hivemind SDK.

Build the ML infrastructure powering production AI models for civil engineering and surveying, from large-scale training data pipelines to real-time inference services. You’ll work across data lineage, model serving, deployment, and tooling for ML engineers.

Waymo is hiring a Senior Machine Learning Engineer to build scalable infrastructure, models, simulation workflows, and evaluation systems for autonomous driving. The role combines distributed ML systems, data engineering, model development, and large-scale driver evaluation.

Lead development of large-scale, interactive web-based visualization tools that help engineers understand AI/ML performance and diagnose inference bottlenecks. This staff-level role bridges frontend engineering, profiling data, developer tooling, and computer architecture.

Lead applied research engineering for next-generation game-world authoring, turning computer vision and machine learning ideas into compelling prototypes and engine features. This senior role blends technical research, game development, and collaboration with artists and designers.

GPU Kernel Engineer designing and optimizing custom kernels to accelerate large-scale AI training and inference. The role works across low-level GPU programming, ML frameworks, and hardware performance optimization.

Lead the engineering of a real-to-sim stack that helps robotics teams close the sim-to-real gap for embodied AI. This hands-on Staff Engineer role combines robot learning, simulation, evaluation, customer insight, and technical leadership.

Build and operate end-to-end AI/ML systems that support mission-critical federal government and defense solutions. This hybrid role focuses on production-grade models, MLOps, and reliable AI/ML pipelines across text, image, and video data.