Machine Learning Engineer, AI Powered: Custom Models

The Custom Models team at GitLab is responsible for allowing customers to deploy and customize the outputs of Generative AI models and fine-tune models for use within the GitLab product and beyond. As a Machine Learning Engineer in this team, you'll work on developing improvements to models to generate new content using machine learning models in a secure, well-tested, and performant way. You'll collaborate with product managers, engineers, and other stakeholders as a machine learning specialist, advocating for improvements to product quality, security, and performance.

Key responsibilities include:

• Developing improvements to models for generating new content
• Working with complex data for feature development using machine learning models
• Collaborating with various stakeholders as a machine learning specialist
• Solving technical problems of moderate scope and complexity
• Crafting code that meets internal standards for style, maintainability, and best practices
• Shipping small features and improvements independently, and collaborating on larger projects
• Participating in code reviews and engineering projects
• Participating in on-call rotations for troubleshooting and urgent issues

Required qualifications:

• Master's degree with 2+ years of ML experience or PhD in Machine Learning or Data Science
• Professional experience with Python
• Experience with performance optimization and problem-solving
• Comfort working in an agile, iterative software development process
• Strong communication and organizational skills
• Alignment with GitLab's values and ability to thrive in a remote environment

Desired qualifications:

• Experience with prompt engineering and Retrieval Augmented Generation (RAG)
• Experience building, training, and implementing deep learning models
• Experience with deep learning frameworks like PyTorch or TensorFlow
• Professional experience fine-tuning LLMs
• Experience with MLOps infrastructure

This role offers the opportunity to shape the future of AI at GitLab, working with a global team in a fully remote environment. GitLab provides comprehensive benefits, including flexible paid time off, equity compensation, and growth and development opportunities.

The base salary range for this role is $112,000 - $240,000 USD for candidates in California, Colorado, Hawaii, New Jersey, New York, Washington, and DC.