Current Openings


Computer Vision Engineer

Apply

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

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

 


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

 


Senior Software Engineer - Machine Learning

Apply

Petuum, Inc. seeks a Senior Software Engineer (Machine Learning) in Sunnyvale, CA responsible for providing engineering leadership, guidance, and direction with respect to the design/development and implementation of new machine learning (ml) platforms, models, and prototypes in support of the development of Petuum's next generation artificial intelligence (AI) software product(s).

Responsibilities

  • serving as a technical lead on the machine learning (ML) engineering team with respect to the design, implementation and evaluation of new ML software models
  • defining and ensuring adherence to the ML product development strategy, standards, policies, tools and procedures during the entire software development life cycle
  • driving quality practices, standards and metrics to achieve high levels of service availability and customer satisfaction; (iv) ensuring that ML projects, department milestones, and goals are met while adhering to approved budgets
  • developing, implementing, and enhancing statistical/ML algorithms for integration into software development projects; (vi) designing machine learning methods for specific applications, using bias-variance tradeoff, exploration/exploitation and various model families (i.e., Neural Net, decision trees, Bayesian models, LSTM, instance-based learning, association learning, and deep learning algorithms)
  • developing software algorithms in accordance with customized ai solutions
  • implementing numerical methods and data visualizations into software prototypes
  • building software code architectures for implementation across various software development projects
  • architecting, maintaining, and updating software libraries (i.e., Kmeans, MLR, and C++), numerical methods, and data visualizations
  • facilitating the code review process and mentoring junior software engineers (ML) with respect to the same
  • keeping abreast of emerging ml technology trends and reporting on/presenting cutting-edge software developments to cross-functional engineering and product management teams to ensure the integration of market-specific product designs.

Qualifications

  • Master’s degree in Data Science, Data Informatics, Computer Science, Computer Engineering, or a directly related field plus five (5) years of experience in a related position.
  • Five (5) years of experience designing/developing software, building scalable systems, and managing software development teams, including five (5) years with each of the following: (i) UNIX OS; (ii) C, Java, or Python; and (iii) relational databases.
Apply  |   Top

 


Senior Product Manager/Group Product Manager – Enterprise AI/ML Platform and Open-Source Toolkits

Apply

We are looking for a strong technical product manager who is passionate about machine learning and AI to lead the development of our advanced AI platform. You will need to understand how to translate our industry leading CASL open-source software into enterprise-ready AI/ML platform for customers. You should be equally comfortable discussing the deep technical machine learning details with our engineering team as well with explaining the customer level benefits of these technologies. You will need to help guide our engineering and marketing teams to build new product features based on market analysis, customer needs, and business dynamics to build the best enterprise AI/ML platform for our customers.

Team

Product management plays a leading role at Petuum and is responsible for envisioning, designing, developing, and helping to build a state-of-the-art AI/ML platform and machine learning tools for customers. Part of the responsibilities for the product management team will be working with our world-renowned research team to productize their cutting-edge research and open-source development into enterprise software. Petuum’s product team will need to understand customer’s needs and develop a roadmap to build out the enterprise AI/ML platform by prioritizing, collaborating, and influencing across a variety of Petuum's different organizational functions to achieve the goal of productization.

Responsibilities

  • Drive AI/ML platform product roadmap from requirements to productization. Establish product goals and develop roadmap.
  • Govern product development efforts, including research, requirements definition, metrics analysis, technical specifications, development, and launch efforts.
  • Translate our CASL research and open-source software into enterprise AI/ML platform for customers.
  • Advocate for the greater adoption of Petuum’s CASL open-source offering.
  • Create and review product specifications and prioritization of requirements to align customer needs and internal resources.
  • Work with Go-To-Market & Sales teams to position the product, define pricing, and product packaging.
  • Influence key stakeholders and work with cross-functional teams to ensure successful execution.
  • Provide technical guidance to customer teams using Petuum’s AI/ML software products.
  • Other duties as assigned. 

Qualifications

  • Bachelor of Science in Computer Science, Engineering, Statistics, or similar field and Master of Science or Business Administration degrees
  • 5+ years of experience working on enterprise products as a lead developer, data scientist manager, or product manager.
  • Hands-on experience with machine learning frameworks and techniques.
  • Experience in Data Science/AI/ML/big data technology landscape with at least one successful launch in the market.
  • Experience defining detailed requirements, working closely with highly technical individuals in a Machine Learning engineering organization.
  • Detail-oriented but can absorb/define the bigger picture.
  • Experience in Enterprise software development.
  • Experience with Open-Source is a plus.
  • Prior experience in platform products is a plus.
  • Startup experience is a plus.
Apply Now   |   Back to Top

 


Manager - DevOps

Apply

Petuum is seeking a DevOps manager that has passion for delivering Petuum’s innovations quickly to our customers and with the best quality possible. 

 You are an experienced “hands-on” technical leader who will manage our software tools, hardware and computing resource (both on premises and in cloud) for development and own the continuous integration, delivery and deployment (CI/CD) process and automation. You will hire, mentor, and coach employees to develop a best-in-class DevOps team. The focus is to ensure fast delivery of Petuum’s products and with the best quality possible. You will help to build a DevOps culture with our engineers and continuously drive for high efficiency to serve our customers better. If you are a self-starter that thrives working in new technologies to drive innovation and discovery, you should join our team. 

Responsibilities

  • Manage tools and engineering infrastructure (both hardware and software) to support our development environment.  
    • Drive improvements so that we can test, release and deploy complex, large scale distributed system quickly. 
    • Own the validation and compliance stage to satisfy our customer’s needs in security and compliance. 
    • Plan and execute complex integrations and deployments for high profile customers. 
    • Provide the infrastructure-as-code needed to maintain multiple development/test/stage/production environments. 
  • Hire, mentor, and coach employees to develop a best-in-class DevOps team. 

Work with engineering teams to influence software development best practice and leverage new tools and infrastructure to build repeatable solution and drive for innovation.

Minimum Qualifications

  • Strong analytical and effective communication skills. 
  • Bachelor’s degree in Computer Science or related technical field or equivalent practical work experience. 
  • 7+ years of industry experience and worked on enterprise software testing, automation and infrastructure. 1-2 years managing a DevOps/SRE team. 
  • Strong knowledge of software development methodologies and the software development lifecycle. 
  • Strong understanding of Agile models and methodology. 
  • Expert in scripting (BashPowerShell or similar languages). 
  • Proficient with source code management tools (Git or similar) 
  • Ability to design and support CI/CD pipelines (Jenkins or similar) 
  • Familiarity with docker containers and Kubernetes technology. 
  • Exposure to infrastructure-as-code frameworks such as Terraform and CloudFormation preferred. 
  • Experience with managing or developing or deploying highly-available and distributed software system with multiple repositories preferred. Experience with web-based API/restful services preferred. 
Apply   |   Top

 


Director - People Operations

Apply

Petuum Inc. seeks to employ a Director (People Operations) in Sunnyvale, CA, responsible for planning, directing, and coordinating the company’s people management operations, including all activities/initiatives related to Petuum’s Human Resources (HR) and Recruitment functions in all of our offices (including China).

Responsibilities

  • leading the strategic direction of the People function while overseeing and continuously improving the entire employee journey including hiring, onboarding, culture, performance management, talent development, and other people initiatives for all Petuum employees across offices in Pittsburgh, PA, Sunnyvale, California, and China;
  • building a robust recruiting operations engine to ensure that recruitment/hiring goals are met within the targeted time and are appropriately budgeted for to ensure optimal productivity, as well as a world-class candidate experience and interviewer experience; 
  • designing and implementing performance evaluation and management processes and systems to track divisional objectives and key results (OKR) progress, address HR issues with senior-level management, and implement real-time problem solving of personnel issues;
  • conducting benchmarking analyses of Petuum’s People Operations processes and employee benefits structures to ensure we remain competitive in the market and making recommendations for new employee benefits initiatives;
  • managing organizational compensation structures, including monitoring and analyzing competitors’ market data on a regular basis;
  • partnering with management to develop and implement organization-wide learning curriculum and content for leaders and employees;
  • implementing and overseeing the maintenance of appropriate HR policies and procedures across all business functions/units and offices, including developing and maintaining HR policy trackers and relevant data sets and recommending needed changes
  • developing exceptional cultural and onboarding initiatives/experiences for all offices (with respect to employee branding, workplace experience, etc.);
  • setting metrics for the People Management function while driving transparency in the process to identify opportunities to increase efficiency;
  • establishing workforce analytics structures to keep all business units informed of key HR and recruiting metrics (i.e., cost to hire, time to hire, cost of attrition, etc.);
  • leading inclusion and diversity as well as community engagement initiatives.

Qualifications

  • Must have a Master’s degree in Human Resources or Business Administration.
  • Must be a member of SHRM or HRCI and have five (5) years of experience in the People Operations or Human Resources field and in at least one (1) of these People Management functions (i.e., end-to-end recruiting, compensation & benefits, or talent management).
  • Must also have three (3) years of experience with each of the following: (i) designing and leading People Management strategies and initiatives with proven results for hiring, promotion, and retention; (ii) managing and partnering with international teams in different locations outside of the US (i.e., Asian or European market) and leading international People Organization’s roadmap and strategy to support the growth of the business and HR initiatives; (iii) coaching/mentoring and fostering relationships across all levels; and (iv) program and organizational management.
  • Must be fluent in Mandarin Chinese language.

Apply   |   Top

We're also hiring for our industrial team.  See those roles here.