Software Engineer - Model Training Infrastructure - USDS

About the team

The mission of our AML team is to push the next-generation AI infrastructure and recommendation platform for the ads ranking, search ranking, live & ecom ranking in our company. We also drive substantial impact on core businesses of the company. Currently, we are looking for Machine Learning Engineer - Model Training Infrastructure to join our team to support and advance that mission. - Responsible for the design and implementation of a global-scale machine learning system for feeds, ads and search ranking models.
- Responsible for improving use-ability and flexibility of the machine learning infrastructure.
- Responsible for improving the workflow of model training and serving, data pipelines, storage system and resource management for multi-tenancy machine learning systems.
- Responsible for designing and developing key components of ML infrastructure and mentoring interns.

Minimum Qualifications
- Bachelor's/Master's degree in Computer Science, Computer Engineering, or related fields or equivalent years of experience in a software engineering role
- Proficient in C/C++/CUDA/Python, and have solid programming skills.
- Familiar with deep learning frameworks (TensorFlow/Pytorch).
- Experience in developing and deploying large-scale systems. Preferred Qualifications
- Ability to work independently and complete projects from beginning to end and in a timely manner.
- Good communication and teamwork skills to clearly communicate technical concepts with other teammates.
- Experience on improving core machine learning infrastructure(TensorFlow, Pytorch, and Jax).
- Experience contributing to an open sourced machine learning framework (TensorFlow/PyTorch).
- Experience in using/designing open-source machine learning lifecycle management systems: TFX As a condition of employment, all successful candidates must be able to establish authorization to work in the United States. For this position, the Company does not provide sponsorship for any immigration-related benefits.