Current Openings


Natural Language Processing Engineer

Apply

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. You will be responsible for building robust, effective, and well-packaged modern Natural Language Processing techniques, as well as contributing to our CASL open source projects.

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 and text generation.
  • 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   |   Top

 


ML Systems and AutoML Engineer

Apply

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. 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.

Responsibilities

  • 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.
  • Develop parallel programming techniques to simplify distributed ML programming.
  • 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

  • 1+ years of experience in one or more areas listed below:
    • AutoML areas such as hyperparameter tuning, architecture search or manual design, data preparation, augmentation, or feature engineering
    • Distributed systems
    • Network communication, or storage systems
  • Hands-on experience with at least one popular deep learning framework 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 industry/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.
  • 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.
Apply   |    Top

 


Software Engineer - FrontEnd UI

Apply

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 FrontEnd UI engineers who can deliver consistently in a fast-paced and high-quality manner. You 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, as well as contribute to our CASL open source projects.

Responsibilities

  • Work with Product and Engineering team to build an incredible and creative UI for our products
  • Design, implement and test efficient, reusable UI components
  • Learn quickly about AI & ML concepts and workflow
  • Clearly communicate your work to a broader audience through demos, presentations, and blog posts

Minimum Qualifications

  • Bachelor's Degree in Computer Science or related quantitative field
  • Expert level development skills in modern JavaScript (ES6 and TypeScript), HTML5, SCSS/CSS
  • 1+ years of experience in developing applications with modern JavaScript frameworks/libraries such as Angular, React
  • 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
  • 4+ years developing large scale single page web applications with significant client-side contributions
  • Prior experience using D3, RxJS, Redux
  • Experience with REST or GraphQL
  • Good working knowledge of Python
  • Familiarity with a backend web framework like Express/Node.js, Django, Rails, Flask, Spring Boot
  • Startup experience, interest in AI Machine Learning, open-source contributions
Apply   |   Top

 


Software Engineer - BackEnd

Apply

The Petuum team is looking for talented, motivated full-time BackEnd Engineers who can consistently deliver with high-quality in a fast-paced startup environment. You will be responsible for building robust, effective, and well-packaged modern distributed machine learning platform services, as well as contributing to our CASL open source projects.

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   |   Top

 


Software Engineer - BackEnd [Pittsburgh]

Apply

Petuum Inc. seeks a Software Engineer (Back-End) in Pittsburgh, PA responsible for building machine learning (ML), computer vision, and natural language processing (NLP) software models and tools in support of the development of Petuum’s next-generation artificial intelligence (AI) software product(s).

Responsibilities

Specific job duties include:

  • architecting, designing, implementing, and testing back-end data and ML pipelines/services
  • developing distributed system, ML, database, and high-performance computing software in support of Petuum’s internal ML software library
  • building raw-process data tools using Tensorflow or PyTorch for implementation into computer vision and NLP software applications
  • implementing next-generation data transformation and ML software systems into newly developed applications
  • applying ML models to prepared software data during the testing process
  • designing and implementing reusable and testable software system libraries, frameworks, and services
  • architecting and implementing software libraries (using SQL, Java, C/C++, etc.), numerical methods, and data visualizations
  • conducting performance testing with respect to scalability, convergence, speed and throughput
  • implementing and evaluating ML/AI micro-service architectures, versions controls and CI/CD processes while supporting high-performance codebase standards
  • keeping abreast of and reporting on emerging ML software system trends/developments as well as implementing updates into existing ML, computer vision, and NLP pipelines for various ML software prototypes

Qualifications

  • Must have a Master’s degree in Computer Science, Information Science/Systems, or a directly related field.
  • Must have any experience with or knowledge of (through any completed university-level coursework, seminars, workshops, or real-world, hands-on experience) each the following:
    • ETL workflow or Data pipeline and process GigaBytes
    • SQL database (i.e., PostgreSQL or MySQL)
    • C, C++, or Python; (iv) Bash
    • any flavor of Linux operating system for development platform
    • data libraries (i.e., NumPy or Pandas)
    • Machine Learning libraries (i.e., scikit-learn or OpenCV)
    • Machine Learning tools (i.e., TensorFlow or PyTorch)
Apply   |   Top