1,446 PyTorch jobs
Open roles that use PyTorch for machine learning.
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…
Member of Technical Staff ML Product Engineering
You will develop and optimize models for production use cases, translate customer needs into technical solutions, implement post-training approaches and agentic workflows, work on data and evaluation pipelines,…
Software Engineer ML Developer Experience
You will build developer tooling and MLOps capabilities across CLI, SDK, API, MCP, cloud infrastructure, and the Ray runtime. You will improve distributed execution, debugging, observability, production serving, and…
Senior Field Application Engineer
You will serve as a technical expert for customers in India, guide hardware evaluation and deployment, resolve technical issues, support sales and launches, gather product feedback, and create product documentation.…
Software Engineer Applied ML Discovery Recommendation and Search
You will work with product, data science, and data platform partners to design and implement applied ML models and infrastructure. You will build ML backend systems for recommendation, ranking, and search, including…
Machine Learning Scientist ML Research
You will design and conduct experiments on AI model behavior, create metrics and evaluation protocols, and analyze large-scale human preference data. You will prototype research ideas, collaborate to scale findings…
Distributed LLM Inference Engineer
You will deliver end-to-end batch and online inference solutions at scale. You will integrate Ray Data and LLM engines, optimize cost and performance, work with open-source communities, contribute improvements, and…
Member of Technical Staff, RL Infra
You will design and optimize infrastructure for large-scale reinforcement learning and post-training workloads. You will improve distributed pipeline reliability and throughput and create monitoring and observability…
Member of Technical Staff, LLM Evaluation Infra
You will build automated evaluation pipelines for model training and deployment. You will develop benchmarks and metrics, analyze model outputs for failures and biases, and turn real-world use cases into meaningful…
Member of Technical Staff, Inference & Serving
You will build low-latency model serving systems and distributed inference infrastructure. You will manage traffic, autoscaling, versioning, deployments, and observability, while converting model advances into…
Member of Technical Staff, Training Infra
You will design and optimize distributed training systems across large GPU fleets. You will improve training throughput and efficiency and develop reusable frameworks that strengthen reliability, reproducibility, and…
Algorithm AI Research Engineer
You will develop proprietary LLMs for agentic tasks such as coding and tool use. You will conduct joint frontier-AI research with overseas institutions and develop asynchronous RL pipelines for LLM post-training using…