1,427 PyTorch jobs
Open roles that use PyTorch for machine learning.
Member of Technical Staff GPU Performance Engineer
You will write and integrate custom GPU kernels, profile training and inference workflows, and remove performance bottlenecks. You will build correctness and numerical tests, maintain benchmark guardrails, and turn…
Senior Machine Learning Engineer Clinical Team
You will own tissue-class segmentation and labeling models for ultrasound CT clinical analysis. You will adapt models across 2D, 3D, and fused inputs; define training and evaluation pipelines; coordinate labeling…
Software Engineer Machine Learning Engineer Agentic AI
You will design and implement autonomous agent systems that explore, execute, evaluate, and improve engineering solutions. You will develop multi-agent workflows, automated evaluation systems, post-training…
Software Engineer Kernel Programming Model
You will design PyTorch-native integration layers and runtime environments for custom kernels. You will implement virtual ISA abstractions, build programming tools and reference implementations, and support a kernel…
Software Engineer LLM Performance and Evaluation
You will build and maintain LLM benchmarks, evaluation pipelines, dashboards, and reports. You will measure latency, throughput, memory use, utilization, and model accuracy; automate reproducible evaluations;…
Senior Product Manager AI Models
You will own the AI model roadmap, select models, establish model-lab partnerships, and lead launches. You will define quality standards and evaluations, gather customer feedback, create technical marketing assets,…
Member of Technical Staff Applied Research
You will develop and train vision-language models for document processing, build data pipelines for curation and synthetic data, and evaluate or fine-tune models against performance targets. You will improve production…
Member of Technical Staff - Post Training Applied
You will lead text-model post-training engagements from scoping through delivery and evaluation. You will turn customer requirements into workflows, generate and assess text data, run fine-tuning and alignment methods,…
ML Engineer - Personalization & Recommendation Systems
You will architect and build personalization and recommendation systems from the ground up. You will model user preferences, create curated feeds, collaborate on generative-model personalization, and take systems from…
Software Engineer Compiler Front End
You will build compiler front-end pipelines that translate deep learning models into internal representations. You will develop graph optimizations, design extensible model-ingestion systems and kernel-language…
Software Engineer AI Model Enablement and Inference
You will analyze model architectures, implement and optimize TCL kernels, and integrate models into Furiosa-LLM. You will validate correctness on NPUs, automate benchmarking and regression testing, evaluate…
Member of Technical Staff Performance Optimization
You will optimize AI workloads across GPU kernels and distributed systems. You will profile bottlenecks, improve latency and resource efficiency, implement CUDA and Triton optimizations, benchmark performance, and…
Member of Technical Staff LLM Infrastructure
You will design and develop scalable backend infrastructure for distributed training, inference, and data pipelines. You will maintain CI/CD, control-plane, and model-serving systems; improve reliability and…
Member of Technical Staff AI Training Infrastructure
You will design, build, and optimize infrastructure for large-scale model training. You will develop distributed training pipelines, optimize workloads across GPUs and data centers, manage training data storage,…
Member of Technical Staff Cloud Infrastructure
You will architect and build backend infrastructure for distributed training, inference, and data processing. You will design core services, lead technical discussions, optimize compute, storage, and networking,…
Member of Technical Staff Cloud Infrastructure Singapore
You will design and develop core backend systems for high-performance generative AI workloads. You will build scalable and stable software components, conduct design and code reviews, collaborate with cross-functional…
Machine Learning Engineer Reliability
You will own the reliability, security, and safety of production generative media model APIs. You will build observability and safe deployment systems, lead incident response, improve GPU capacity management, and help…
Member of Technical Staff Reinforcement Learning
You will design, develop, and optimize reinforcement-learning training pipelines for diffusion-based LLMs. You will build and evaluate reward models, fine-tune and scale generative AI models, process data, evaluate…
Member of Technical Staff Training
You will develop diffusion-language-model architectures, implement training objectives and loss functions, research controlled generation and multimodal methods, improve training and inference efficiency, and develop…