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 AutoML (automated machine learning) Engineers who are able to deliver consistently in a fast-paced and high-quality manner. You will be responsible for helping build robust, effective, and well-packaged modern AutoML systems, as well as contributing to our CASL open source projects.


  • Collaborate with system architects, designers, and engineers to support the development of robust machine learning systems.
  • Contribute high-quality code and lead efforts in building Petuum’s open-source CASL projects such as AdaptDL, AutoDist, Tuun.
  • Learn and implement state-of-the-art deep AutoML algorithms to support tasks such as hyperparameter optimization, neural architecture search, data augmentation, feature engineering, and more.
  • Assess and recommend technology choices and directions in consideration of cost-benefit trade-offs.
  • Communicate your work to a broader audience through talks, tutorials, and blog posts.

Minimum Qualifications

  • Master’s degree in Computer Science, Machine Learning, or related fields.
  • 5+ years of industry or research experience, including at least 3 years of experience in AutoML areas such as hyperparameter tuning, architecture search or manual design, data preparation, augmentation, or feature engineering
  • Experience with model-based optimization (e.g. Bayesian optimization) methods or software frameworks such as Hyperopt, Optuna.
  • Hands-on experience with at least one popular deep learning framework such as PyTorch and Tensorflow.
  • Strong engineering skills in Python and C++.

Preferred Qualifications

  • Ph.D. degree in Computer Science, Machine Learning, or related fields.
  • Experience in deploying machine learning algorithms in resource-restricted environments such as mobile or embedded systems.
  • Experience in developing with Docker, Kubernetes, Ray, NNI, etc.
  • Experience in contributing to notable open-source ML software, such as TensorFlow, PyTorch, etc.
  • Publication (or submission) of a paper to machine learning or operating systems conferences.

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