Current Openings


Computer Vision Engineer

The Petuum team is looking for talented, motivated full-time Computer Vision (CV) Engineers who are able to deliver consistently in a fast-paced and high-quality manner. This engineer will be responsible for building robust, effective, and well-packaged modern CV techniques. 

Responsibilities

  • Learn and implement state-of-the-art deep learning algorithms to support tasks such as object detection, image classification, and image captioning.
  • Deliver robust, well-tested, and fully documented CV modules at different granularity levels.
  • Continuously improve the efficiency, robustness, and cost of existing modules.
  • Contribute high-quality code and lead efforts in building Petuum’s open-source projects.
  • Collaborate with system architects, designers, and engineers to support the development of robust machine learning systems.
  • Assess and recommend technology choices and directions in consideration of cost-benefit trade-offs.

Minimum Qualifications

  • 2+ years of experience in one or more CV areas such as object detection, pattern recognition, and image classification.
  • Hands-on experience with at least one popular deep learning frameworks such as PyTorch and Tensorflow.
  • High-level engineering skills in Python and C++.

Preferred Qualifications

  • Master’s degree in Computer Science, Machine Learning, or related fields with 2+ years of equivalent practice or research experience, or Ph.D. degree in Computer Science, Machine Learning, or other relevant degrees.
  • Experience in deploying machine learning algorithms in resource-restricted environments such as mobile or embedded systems is highly preferred.
  • Experience with medical image processing is a plus.
  • Experience with multi-modal machine learning.
  • Experience with developing popular open-source projects.

Apply Now


Natural Language Processing Engineer

The Petuum team is looking for talented, motivated full-time Natural Language Processing (NLP) Engineers who are able to deliver consistently in a fast-paced and high-quality manner. This engineer will be responsible for building robust, effective, and well-packaged modern Natural Language Processing techniques. 

Responsibilities

  • Learn and implement state-of-the-art natural language processing models to support tasks such as machine translation, question answering, and knowledge base construction.
  • Deliver robust, well-tested, and fully documented models to support the above-mentioned tasks.
  • Continuously improve efficiency, robustness, and cost of existing modules.
  • Contribute high-quality code and lead efforts in building Petuum’s open-source projects.
  • Collaborate with system architects, designers, and engineers to support the development of robust machine learning systems.
  • Assess and recommend technology choices and directions in consideration of cost-benefit trade-offs.

Minimum Qualifications

  • 2+ years of experience in one or more NLP areas such as Information Extraction, Translation, Text Generation, and Dialogue System.
  • Hands-on experience with at least one current ML deep learning frameworks such as PyTorch and Tensorflow.
  • High-level engineering skills in Python and C++.

Preferred Qualifications

  • Master’s degree in Computer Science or related fields with 2+ years of ML and language technology development or research experience, or Ph.D. degree in Computer Science, Machine Learning, or other relevant degrees.
  • Experience with low-resource language processing is highly preferred.
  • Experiences deploying machine learning systems in resource-restricted environments such as mobile or embedded systems is highly preferred.
  • Experience with multi-modal machine learning.
  • Experience with developing popular open-source projects.

Apply Now


Machine Learning Systems Engineer

The Petuum team is looking for talented, motivated full-time ML System Engineers who can deliver consistently in a fast-paced and high-quality manner. This engineer will be responsible for building robust, effective, and well-packaged modern distributed machine learning systems.

Responsibilities

  • Develop high-quality open-source software to simplify distributed ML programming.
  • Identify, implement, and evaluate architectural improvements to CASL core.
  • Improve the testing process for various CASL projects such as AutoDist, AdaptDL, Tuun to make releases as smooth as possible.
  • Communicate your work to a broader audience through talks, tutorials, and blog posts.

Minimum Qualifications

Experience in one or more areas as listed below:
  • Machine learning algorithms
  • Distributed systems
  • Network communication, or storage systems
  • Machine Learning frameworks including TensorFlow, PyTorch, etc.
  • High-level engineering skills in Python and C++

Preferred Qualifications

  • Master's degree in Computer Science, Machine Learning, or related fields with 2+ years of equivalent practice or research experience, or Ph.D. degree in Computer Science, Machine Learning, or other relevant degrees.
  • Publications in relevant venues such as OSDI, SOSP, ATC, Eurosys, SysML, etc.
  • Experience in deploying machine learning algorithms in resource-restricted environments such as mobile or embedded systems is highly preferred.
  • Experience with developing popular open-source projects.
  • Experience in contributing to notable open-source ML software, such as TensorFlow, PyTorch, etc.

Apply Now


ML Systems and AutoML Engineer

The Petuum team is looking for talented, motivated full-time ML Systems and AutoML (automated machine learning) Engineers who are able to deliver consistently in a fast-paced and high-quality manner. This engineer will be responsible for helping build robust, effective, and well-packaged modern AutoML systems.

Responsibilities

  • 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.
  • Deliver robust, well-tested, and fully documented AutoML modules at different granularity levels.
  • Continuously improve the efficiency, robustness, and cost of existing modules.
  • Contribute high-quality code and lead efforts in building Petuum’s open-source projects. 
  • Collaborate with system architects, designers, and engineers to support the development of robust machine learning systems.
  • Work on applications of AutoML and other systems to real-world ML and optimization tasks.
  • Assess and recommend technology choices and directions in consideration of cost-benefit trade-offs.

Minimum Qualifications

  • 1+ years of experience in one or more AutoML areas such as hyperparameter tuning, architecture search or manual design, data preparation, augmentation, or feature engineering.
  • Hands-on experience with at least one popular deep learning framework such as PyTorch and Tensorflow.
  • High-level engineering skills in Python.

Preferred Qualifications

  • Master’s degree in Computer Science, Machine Learning, or related fields with 2+ years of equivalent practice or research experience, or Ph.D. degree in Computer Science, Machine Learning, or other relevant degrees.
  • Experience with model-based optimization (e.g. Bayesian optimization) methods or software frameworks is a plus.
  • Experience with developing popular open-source projects.
  • Publication (or submission) of a paper to machine learning conferences.

Apply Now

 


Software Engineer - Full Stack

The Petuum team is looking for motivated full-time Full Stack Engineers who are enthusiastic about Artificial Intelligence technologies. This engineer will work with our engineering team to build and maintain sophisticated, powerful user experiences which play important roles in our Machine Learning systems and vertical solutions.

Responsibilities

  • Work with our UX and Product teams to review designs and provide feasibility feedback.
  • Work with Back End engineers and Machine Learning engineers to realize those designs.
  • Write efficient, reusable, testable UI components, such as interactive visualizations of data.
  • Contribute to full-stack web applications.
  • Learn quickly about AI & ML concepts and workflow.
  • Design, implement, and test our front-end services.

Minimum Qualifications

  • Bachelor's Degree in Computer Science or related quantitative field.
  • 1-4+ years of experience in web-development with significant client-side contributions using JavaScript, HTML, SCSS/CSS.
  • 1-4+ years of experience in at least one backend web framework like Express/node.js, Django, Rails, Flask, Spring Boot, etc. and RESTful patterns or GraphQL.
  • Familiarity with a few languages, such as JavaScript, TypeScript, Python, Go.
  • Clear understanding of componentized MVC architecture (and its variations) in modern SPA frameworks. 
Preferred Qualifications
  • An advanced degree in Computer Science or equivalent practical work experience.
  • Startup experience, interest in AI and ML, and open-source contributions.
  • Experience working with any of the following: distributed and parallel systems, machine learning, data visualization. 
  • Experience with more frameworks/Libraries such as Angular, React, Redux, D3.

Apply Now

 


Software Engineer - BackEnd

The Petuum team is looking for a talented, motivated full-time BackEnd Engineer who can consistently deliver with high-quality in a fast-paced startup environment. This engineer will be responsible for building robust, effective, and well-packaged modern distributed machine learning platform services.

Responsibilities

  • 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

Minimum Qualifications

  • Master's degree in Computer Science or related quantitative fields
  • Expertise in distributed systems
  • Experience with the architecture and development of robust services with focus on low latency and scaled throughput
  • Experience with databases, network or storage systems
  • Expert in Python and 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

  • 4+ years of industry experience maintaining high codebase standards in a growing team
  • Experience in a backend framework like Django or similar with REST or GraphQL APIs
  • Development and troubleshooting in Docker and Kubernetes environment
  • 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

Apply Now