Remote Software engineering Jobs · PyTorch

Job listings

  • Design and implement advanced knowledge distillation pipelines, including teacher-student approaches and multi-teacher architectures.
  • Run large-scale machine learning experiments to optimize model quality, latency, efficiency, and cost trade-offs.
  • Collaborate with research teams to transform emerging distillation techniques into reliable production-ready implementations.

Our partner is an innovative company focused on advancing the efficiency and scalability of next-generation machine learning systems. They offer a remote-friendly work environment with an async-first culture and a small, senior team combining research expertise and engineering excellence.

  • Design and scale production ML systems for LLM-based applications.
  • Build training and evaluation pipelines for continuous model improvement.
  • Fine-tune foundation models using modern adaptation techniques such as LoRA, QLoRA, SFT and DPO.

A1 is a new AI venture building the next generation of AI-native productivity applications, starting with an email agent that uses autonomous AI. Backed by an initial $100M investment, the company is a small, high-talent founding engineering team focused on solving challenging AI infrastructure problems.

$216,700–$303,400/yr

  • Independently own high-value optimization initiatives across training, inference, or launch-readiness for important Ads ML workloads.
  • Diagnose bottlenecks in real production systems using profiling, benchmarking, and observability.
  • Build performance tooling, optimization playbooks, and efficiency primitives that benefit multiple teams.

Reddit is a community of communities built on shared interests and authentic conversations. With 100,000+ active communities and approximately 126 million daily active unique visitors, Reddit has a flexible workforce and values collaboration.

  • Own and improve the machine learning models and systems powering the Home feed and Shortcuts experience.
  • Design and build personalized recommendations serving millions of Spotify listeners globally.
  • Train, fine-tune, and optimize large language models using techniques like SFT and distillation.

Spotify is a leading audio streaming platform that delivers personalized music and podcast experiences to millions of listeners globally. The Personalization team is a large, collaborative group behind iconic features like Discover Weekly, with a culture of innovation and user-centricity.

  • Design and train custom object detection models for retail environments.
  • Fine-tune and deploy vision-language models for product understanding and scene reasoning.
  • Optimize models for edge deployment using quantization and pruning techniques.

Panoptyc is building the future of retail intelligence through computer vision and AI. The company has a focused, collaborative team of engineers working on cutting-edge retail applications.

  • Lead the development of advanced computer vision and biometric solutions for identity verification.
  • Design, train, and scale production-grade machine learning systems with end-to-end pipeline ownership.
  • Collaborate with cross-functional teams to build reliable, ethical, and scalable AI systems with global impact.

The partner company develops advanced identity verification and digital trust solutions using AI. It is a growing technology organization with a collaborative and inclusive culture that values diverse perspectives and professional backgrounds.

United States Unlimited PTO

  • Design, develop, and deploy production machine learning models for perception systems, owning end-to-end model development.
  • Analyze model performance, identify failure modes, and improve robustness across diverse driving environments.
  • Collaborate with cross-functional teams to integrate perception models into production autonomy software.

Torc Robotics has been a leader in autonomous driving since 2007, pioneering software for self-driving vehicles. Now part of the Daimler family, they focus on developing software for automated trucks, with a collaborative and energetic culture that values diverse perspectives.

$170,170–$286,000/yr

  • Build and improve ML/CV models, including retraining and optimizing for Samsara-specific problems.
  • Work with petabyte-scale data from camera and sensor devices to develop new models.
  • Partner with firmware and full-stack teams to deploy models for optimal performance and cost.

Samsara is the pioneer of the Connected Operations Cloud, enabling organizations to harness IoT data for actionable insights to improve safety, efficiency, and sustainability. As a public company with over 2.3 million IoT devices deployed globally, Samsara fosters a culture of growth mindset, inclusivity, and teamwork.

  • Lead the design and development of computer vision systems for biometrics, including face attributes, detection, and recognition.
  • Architect, train, and optimize models using PyTorch, TensorFlow, or JAX, and own end-to-end ML pipelines on AWS.
  • Mentor ML engineers and drive technical best practices across the Computer Vision team.

Jumio is a B2B technology company that uses AI and biometrics to prevent online identity fraud and financial crimes. It is a global leader in identity verification with a diverse team and values of Integrity, Diversity, Empowerment, Accountability, and Leading Innovation.

$216,700–$303,400/yr

  • Design, develop, and deploy ML models, including large language models, for various NLP tasks.
  • Collaborate with cross-functional teams to gather requirements, define architectures, and iterate on model development.
  • Stay up-to-date with latest research and contribute to best practices for responsible ML development.

Reddit is a community of communities built on shared interests, passion, and trust, hosting the most open and authentic conversations on the internet. With over 100,000 active communities and approximately 126 million daily active users, it is one of the internet's largest sources of information, fostering a culture of authenticity and community.