Machine Learning Engineer, Tech Lead - Engineering Efficiency and AI Code Assistant

$224,000 - $410,000/Yr

Tiktok - San Jose, CA

Full-time - Mid Level

San Jose, CA

Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services

About the position

As a Machine Learning Engineer (MLE) specializing in code large language models at TikTok, you will play a pivotal role in developing, enhancing, and applying AI-powered models specifically designed for code-related tasks. Your primary focus will be on leveraging advanced code intelligence models to perform comprehensive code reviews, generate unit tests, and implement effective code fixes. This position is integral to improving software engineering processes and ensuring high-quality solutions are delivered to our users. In this role, you will be responsible for designing, training, and optimizing code-specific large language models (LLMs) that address various tasks such as code review, unit testing, and code fixing. You will utilize your expertise in code-centric AI to drive innovation and enhance the overall quality of our software products. Your work will involve collaborating closely with software engineers, data scientists, and other stakeholders to understand their requirements and deliver effective AI-powered tools and solutions. Additionally, you will stay up-to-date with the latest advancements in code LLMs and related AI technologies, conducting research to identify new techniques and methodologies that can be integrated into our solutions. Documenting model development processes, performance metrics, and results will also be a key part of your responsibilities, as you provide detailed reports and insights on model performance and improvements. This position offers a unique opportunity to contribute to TikTok's mission of inspiring creativity and bringing joy through innovative technology solutions.

Responsibilities

• Develop and Enhance Code LLMs: Design, train, and optimize code-specific LLMs to address tasks such as code review, unit testing, code fixing, and other code-centric activities.
• Code Review and Analysis: Utilize code intelligence models to perform thorough and accurate code reviews, identifying potential issues, improvements, and areas for optimization.
• Unit Test Generation: Develop and implement models that automatically generate unit tests to ensure code quality and reliability.
• Code Fixing and Refactoring: Leverage AI models to suggest and apply code fixes, improvements, and refactorings to enhance code performance and maintainability.
• Research and Innovation: Stay up-to-date with the latest advancements in code LLMs and related AI technologies. Conduct research to identify new techniques and methodologies that can be integrated into our solutions.
• Work closely with software engineers, data scientists, and other stakeholders to understand requirements and deliver effective AI-powered tools and solutions.
• Document model development processes, performance metrics, and results. Provide detailed reports and insights on model performance and improvements.

Requirements

• Proven experience in developing and working with code-specific large language models (e.g., GitHub Copilot, OpenAI Codex, DeepMind AlphaCode) and related AI technologies.
• Strong proficiency in programming languages such as GoLang (preferred), Python, Java, or C++. Experience with machine learning frameworks and libraries (e.g., TensorFlow, PyTorch).
• Ability to solve complex problems comprehensively and develop innovative solutions using code-centric AI models.
• In-depth understanding of software engineering principles, code review processes, unit testing methodologies, and code optimization techniques.
• Excellent verbal and written communication skills, with the ability to clearly convey technical concepts and collaborate effectively with cross-functional teams.
• Bachelor's or Master's degree in Computer Science, Engineering, or a related field, or equivalent practical experience.

Benefits

• 100% premium coverage for employee medical insurance
• Approximately 75% premium coverage for dependents
• Health Savings Account (HSA) with company match
• Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life and AD&D insurance plans
• Flexible Spending Account (FSA) Options like Health Care, Limited Purpose and Dependent Care
• 10 paid holidays per year
• 17 days of Paid Personal Time Off (PPTO)
• 10 paid sick days per year
• 12 weeks of paid Parental leave
• 8 weeks of paid Supplemental Disability
• Mental and emotional health benefits through EAP and Lyra
• 401K company match
• Gym and cellphone service reimbursements