Software Engineer Intern (Applied Machine Learning-Engine) - 2025 Summer/Fall (PhD)

$156,000 - $156,000/Yr

Bytedance - San Jose, CA

Full-time - Intern

San Jose, CA

Professional, Scientific, and Technical Services

About the position

ByteDance is seeking a Software Engineer Intern for the Applied Machine Learning-Engine team for the Summer/Fall of 2025. Founded in 2012, ByteDance's mission is to inspire creativity and enrich life through its diverse suite of products, including TikTok, Helo, and Resso. The company is dedicated to fostering an environment where creativity thrives, and every challenge is viewed as an opportunity for growth and innovation. The AML team focuses on developing next-generation machine learning algorithms and platforms that enhance the recommendation system, ads ranking, and search ranking, significantly impacting the company's core businesses. Internships at ByteDance are designed to provide students with hands-on experience and industry exposure, allowing them to turn their ambitions into reality. The internship will last between 12 to 24 weeks, starting in either May/June or August/September 2025. Successful candidates will be responsible for the design and implementation of a global-scale machine learning system, improving the usability and flexibility of the machine learning infrastructure, and enhancing workflows for model training, serving, and resource management. Interns will also have the opportunity to participate in social events and development workshops, laying a strong foundation for personal and professional growth. Candidates are encouraged to apply early, as applications will be reviewed on a rolling basis. The internship offers a unique opportunity to utilize knowledge in real-world scenarios while contributing to impactful projects within the company. Candidates must clearly state their availability in their resumes, including start and end dates, and those who pass the resume evaluation will be invited to participate in a technical online assessment.

Responsibilities

• Responsible for the design and implementation of a global-scale machine learning system for feeds, ads and search ranking models.
• Responsible for improving usability and flexibility of the machine learning infrastructure.
• Responsible for improving the workflow of model training and serving, data pipelines and resource management for multi-tenancy machine learning systems.
• Responsible for designing and developing components of ML infrastructure.

Requirements

• Currently pursuing a PhD in Software Development, Computer Science, Computer Engineering, or a related technical discipline.
• Proficient in C/C++/Python/Golang, and have solid programming skills (e.g. algorithms and data structures).
• Familiar with deep learning frameworks (TensorFlow/PyTorch).
• 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.

Nice-to-haves

• Experience contributing to an open sourced machine learning framework (TensorFlow/PyTorch), experience on improving core machine learning infrastructure.
• Experience in big data frameworks (e.g., K8s/Spark/Hadoop/Flink), experience in resource management and task scheduling for large scale distributed systems, experience in building solutions with AWS, GCP, Azures, OCI, AliCloud or other cloud services.
• Strong background in one of the following fields: Hardware-Software Co-Design, High Performance Computing, ML Hardware Acceleration (e.g., GPU/TPU/RDMA) or ML for Systems.
• Experience in developing and deploying large-scale systems (e.g. Monitoring, Analyzing, Troubleshooting, and Notification systems), strong understanding of code optimizing, routine task automation and failure self-healing, familiar with IaC technologies like Terraform/Ansible.

Benefits

• Hourly rate of $75.
• 100% premium coverage for Full-Time intern medical insurance after 90 days from the date of hire (medical coverage only).
• Paid holidays and paid sick leave, with sick leave entitlement based on the time of joining.
• Mental and emotional health benefits through the Employee Assistance Program.
• Reimbursements for mobile phone expenses.