Backend Engineer - AML Framework Development (Search, Ads, and Recommendation Direction) Singap[...]
Backend engineer on ByteDance's machine learning (AML) framework team, designing and building the Python-layer ML framework powering search, ads, and recommendation — including model encapsulation, optimization tooling, and performance analysis components. Core stack: Python, C++, and deep learning frameworks like TensorFlow/PyTorch.
- Compensation
- Not disclosed
- Location
- Singapore, Singapore
- Work setup
- Not specified
- Seniority
- Not specified
- Employment
- Not specified
- Experience
- 3+ years
- Category
- Backend
- Posted
Skills
Must have
- Bachelor's degree in Computer Science or equivalent with 3+ years of relevant experience
- Familiar with at least one deep learning framework (TensorFlow, PyTorch, or Keras) and automatic differentiation implementation
- Familiar with DNN, CNN, LSTM and common model architectures in search and recommendation
- Excellent programming skills, solid data structures knowledge, proficient in Python, familiar with C++
- Outstanding framework design and interface encapsulation abilities, knowledge of common design patterns
Nice to have
- Ability to thrive in a fast-paced environment
- Experience in overall design and optimization of distributed training projects
- Familiarity with model computation efficiency optimization for mainstream deep learning frameworks
Full description
Backend Engineer - AML Framework Development (Search, Ads, and Recommendation Direction)
Location:
Team:
Technology
Employment Type: Regular
Job Code: A15051
Responsibilities
- Framework Design, Module Encapsulation/Development, and Model Optimization Tools Development
- Responsible for the design and implementation of the machine learning framework at the Python layer
- Responsible for the general and efficient encapsulation of model structures at the Layer/Module level
- Responsible for the design and development of components for model performance and effectiveness analysis
- Support the research and implementation of model algorithms related to various business areas within the company
Qualifications
Minimum Qualifications
- Bachelor’s degree in Computer Science or equivalent with 3+ years of relevant experience
- Familiar with at least one deep learning framework, such as TensorFlow, PyTorch, or Keras, and knowledgeable about automatic differentiation implementation
- Familiar with basic network models such as DNN, CNN, and LSTM, and understand commonly used model architectures in search and recommendation
- Excellent programming skills, solid understanding of basic data structures, proficient in Python, and familiar with C++
- Outstanding framework design and interface encapsulation abilities, with knowledge of common design patterns
Preferred Qualifications
- Ability to thrive in a fast-paced environment
- Experience in the overall design and optimization of distributed training projects
- Familiarity with model computation efficiency optimization solutions for mainstream deep learning frameworks
Founded in 2012, ByteDance’s mission is to inspire creativity and enrich life. With more than a dozen products, including TikTok, Lemon8, CapCut and Pico, ByteDance connects people to content and each other around the world.
ByteDance is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. We celebrate diverse voices and aim to build a workplace that reflects the communities we serve.