PyTorch Software Engineer Jobs

Own the full lifecycle of detection and tracking models powering Helsing’s autonomous drone products, from data curation and training to production deployment on edge platforms. Apply practical machine learning and computer vision expertise to build reliable capabilities for mission-critical defense systems.

Partner hands-on with AI teams to deepen CoreWeave adoption, solve complex production challenges, and scale LLM workloads across the platform. This customer-facing role combines solutions architecture, AI implementation, technical advisory work, and expansion support.

Lead Roku’s machine learning strategy for its Content Platform, building scalable production models and integrating generative AI and vision capabilities. You’ll connect deep AI research with rapid product delivery that improves content discovery and engagement.

Lead the architecture and delivery of secure, production-scale generative AI and machine learning systems at Okta. You’ll build LLMOps, RAG, evaluation, agentic application, and platform capabilities while mentoring engineers and shaping the company’s GenAI strategy.

Build the ML infrastructure that powers Nimble’s autonomous warehouse robots, from highly utilized GPU training clusters to low-latency production inference. This senior/staff-level role combines distributed systems, CUDA optimization, Kubernetes, and robotics-focused machine learning infrastructure.

Build production-scale AI systems that power Axon’s public safety products across cloud and edge environments. This senior role combines machine learning, cloud infrastructure, large-model optimization, and technical leadership.

Build and productionize machine learning models and large-scale systems powering underwriting and portfolio management for Stripe Capital. You’ll work across experimentation, data pipelines, deployment, and product collaboration to deliver scalable ML solutions.

Build the systems, datasets, environments, and publishing tools that power reproducible AI model evaluations and benchmarks. You’ll partner with researchers to turn evaluation methodologies into reliable pipelines, dashboards, and analysis workflows.