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 enhancing and applying AI-powered models specifically designed for code-related tasks. Your primary responsibilities will include developing, training, and optimizing code-specific large language models (LLMs) that address various tasks such as code reviews, unit testing, and code fixing. You will leverage your expertise in code-centric AI to drive innovation and deliver high-quality solutions that improve software engineering processes. In this role, you will utilize advanced code intelligence models to perform comprehensive code reviews, identifying potential issues and areas for optimization. You will also be responsible for developing models that automatically generate unit tests, ensuring code quality and reliability. Additionally, you will apply AI models to suggest and implement code fixes and refactorings, enhancing code performance and maintainability. Staying up-to-date with the latest advancements in code LLMs and related AI technologies will be crucial, as you will conduct research to identify new techniques that can be integrated into our solutions. Collaboration is key in this position; you will work closely with software engineers, data scientists, and other stakeholders to understand requirements and deliver effective AI-powered tools and solutions. Documenting model development processes, performance metrics, and results will also be part of your responsibilities, along with providing detailed reports and insights on model performance and improvements.

Responsibilities

• Develop and enhance code-specific large language models (LLMs) for tasks such as code review, unit testing, and code fixing.
• Utilize code intelligence models to perform thorough and accurate code reviews, identifying potential issues and areas for optimization.
• Develop and implement models that automatically generate unit tests to ensure code quality and reliability.
• Leverage AI models to suggest and apply code fixes, improvements, and refactorings to enhance code performance and maintainability.
• Stay up-to-date with the latest advancements in code LLMs and related AI technologies.
• Conduct research to identify new techniques and methodologies for integration into solutions.
• Collaborate 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).
• 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

• 401(k) matching
• AD&D insurance
• Dental insurance
• Disability insurance
• Employee assistance program
• Flexible spending account
• Health insurance
• 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
• Gym and cellphone service reimbursements