Petuum’s mission is to unlock human productivity and well-being by advancing the limits of AI technology standards and engineering to build trustworthy AI products. The Petuum team is looking for talented, motivated full-time BackEnd engineers who can deliver consistently in a fast-paced and high-quality manner. You will be responsible for building robust, effective, and well-packaged modern distributed machine learning systems, as well as contribute to our CASL open-source projects.


  • Architect, design, implement, test and debug backend data and ML pipelines/services
  • Work with Product and Engineering teams to build new features
  • Write efficient, reusable, scalable, testable backend libraries, and services
  • Learn quickly about AI & ML concepts
  • Contribute high-quality open-source software to simplify distributed ML programming
  • Clearly communicate your work to a broader audience through demos, presentations, and blog posts
  • Mentor junior engineers and facilitate development processes

Minimum Qualifications

  • Bachelor's degree in Computer Science or related quantitative fields
  • Expertise in distributed systems
  • Minimum 3 years of industry experience, including at least 1 year of experience with the architecture and development of robust services with focus on low latency and scaled throughput
  • Experience with databases, network, or storage systems
  • Development and troubleshooting in Docker and Kubernetes environment
  • Expert in Python, good working knowledge of object-oriented programming using C++ or Java
  • Great understanding of version control and CI/CD process
  • Demonstrate strong software engineering practices including code reviews, writing automated unit tests, and product documentation

Preferred Qualifications

  • Experience maintaining high codebase standards in a growing team
  • Experience in a backend framework like Django or similar with REST or GraphQL APIs
  • Experience with popular databases like MongoDB or Postgres
  • Experience with microservice architecture and versioned APIs
  • Exposure to messaging PubSub systems like Pulsar or Kafka
  • Contributed to or owned customization of any popular open-source projects
  • Knowledge of TensorFlow or PyTorch framework

What We Offer

Petuum offers Medical, Dental, Vision, Life/Disability, Paid Time Off, Parental Leave, and more.

Petuum is a welcoming workplace that considers applicants for employment without regard to, and does not discriminate on the basis of, gender, race, protected veteran status, disability, or any other legally protected status. Petuum is an at-will employer.

Apply for this Job

* Required