GitLab Logo

GitLab

Intermediate Machine Learning Engineer, AI Powered: AI Framework

Posted 10 Hours Ago
Easy Apply
Remote
Hiring Remotely in United States
Mid level
Easy Apply
Remote
Hiring Remotely in United States
Mid level
As an Intermediate Machine Learning Engineer at GitLab, you will design and implement evaluation systems for AI models, streamline validation processes, and collaborate with feature teams to enhance AI capabilities. Your role involves building scalable pipelines, managing datasets, and ensuring high-quality AI tools while working in a fully remote, asynchronous environment.
The summary above was generated by AI

GitLab is an open core software company that develops the most comprehensive AI-powered DevSecOps Platform, used by more than 100,000 organizations. Our mission is to enable everyone to contribute to and co-create the software that powers our world. When everyone can contribute, consumers become contributors, significantly accelerating the rate of human progress. This mission is integral to our culture, influencing how we hire, build products, and lead our industry. We make this possible at GitLab by running our operations on our product and staying aligned with our values. Learn more about Life at GitLab.

An overview of this role

Are you passionate about building robust frameworks to evaluate and ensure the reliability of AI models? As a Machine Learning Engineer on GitLab’s AI framework team, you’ll play a critical role in shaping the future of AI-powered features at GitLab. This is an exciting opportunity to work on impactful projects that directly influence the quality of GitLab’s AI capabilities.

You’ll help merge cutting-edge evaluation tools, optimize dataset management, and scale our validation infrastructure. Working closely with other AI feature teams, you’ll ensure that every AI feature we deliver is robust, reliable, and meets the highest quality standards.

Some challenges in this role include designing scalable solutions for LLM evaluation, consolidating disparate validation tools, and contributing to GitLab’s innovative AI roadmap.

Some examples of our projects: 

Consolidating Evaluation Tooling | The GitLab HandbookGitLab.org / AI Powered / ELI5

  • GitLab.org / ModelOps / AI Model Validation and Research / AI Evaluation / Prompt Library 

What You’ll Do  

  • Design and implement technical evaluators for LLM assessment.
  • Contribute to evaluation infrastructure consolidation efforts.
  • Build scalable evaluation pipelines and frameworks.
  • Develop and manage datasets and evaluation metrics.
  • Collaborate with feature teams to integrate validation solutions.
  • Optimize performance across ML evaluation systems.
  • Support improvements to GitLab’s AI-powered tools through validation.
  • Ensure all solutions align with GitLab’s infrastructure and security protocols.

What You’ll Bring 

  • Proven experience designing and implementing LLM evaluation systems.
  • Strong understanding of ML model architectures, including public vs. private implementations.
  • Expertise in ML evaluation metrics and dataset management.
  • Demonstrated ability to build production-grade ML infrastructure.
  • Practical experience with Python-based ML frameworks and evaluation tools (e.g., Langsmith, ELI5).
  • Excellent problem-solving skills with an engineering mindset.
  • Ability to collaborate in an asynchronous, remote-first environment.
  • Familiarity with open-source development and contribution is a plus.

About the team

The AIF team ensures that AI models across GitLab are reliable and well-validated. We focus on building robust evaluation frameworks, consolidating tools, and streamlining processes to scale validation efforts across GitLab’s AI infrastructure. Working on high-impact projects, the team partners with AI feature teams to deliver quality-focused solutions that enhance user trust and product performance.

How GitLab will support you

  • Benefits to support your health, finances, and well-being
  • All remote, asynchronous work environment
  • Flexible Paid Time Off
  • Team Member Resource Groups
  • Equity Compensation & Employee Stock Purchase Plan
  • Growth and development budget 
  • Parental leave 
  • Home office support

Please note that we welcome interest from candidates with varying levels of experience; many successful candidates do not meet every single requirement. Additionally, studies have shown that people from underrepresented groups are less likely to apply to a job unless they meet every single qualification. If you're excited about this role, please apply and allow our recruiters to assess your application.

 

Remote-Global

Country Hiring Guidelines: GitLab hires new team members in countries around the world. All of our roles are remote, however some roles may carry specific location-based eligibility requirements. Our Talent Acquisition team can help answer any questions about location after starting the recruiting process.  

Privacy Policy: Please review our Recruitment Privacy Policy. Your privacy is important to us.

GitLab is proud to be an equal opportunity workplace and is an affirmative action employer. GitLab’s policies and practices relating to recruitment, employment, career development and advancement, promotion, and retirement are based solely on merit, regardless of race, color, religion, ancestry, sex (including pregnancy, lactation, sexual orientation, gender identity, or gender expression), national origin, age, citizenship, marital status, mental or physical disability, genetic information (including family medical history), discharge status from the military, protected veteran status (which includes disabled veterans, recently separated veterans, active duty wartime or campaign badge veterans, and Armed Forces service medal veterans), or any other basis protected by law. GitLab will not tolerate discrimination or harassment based on any of these characteristics. See also GitLab’s EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know during the recruiting process.

Top Skills

Python

Similar Jobs at GitLab

An Hour Ago
Easy Apply
Remote
3 Locations
Easy Apply
Senior level
Senior level
Cloud • Security • Software • Cybersecurity • Automation
The Field CTO at GitLab will lead strategic initiatives by collaborating with customers and internal teams, providing executive-level thought leadership on DevSecOps transformation and solution recommendations. This role includes building relationships with customers, curating field feedback, and driving solution strategies, along with staying updated on emerging technologies and industry trends.
Top Skills: Gitlab
An Hour Ago
Easy Apply
Remote
US
Easy Apply
Entry level
Entry level
Cloud • Security • Software • Cybersecurity • Automation
As an Intermediate Site Reliability Engineer at GitLab, you'll automate workflows and maintain reliability across GitLab environments, respond to user emergencies, build monitoring systems, and collaborate with various teams to enhance operational practices.
Top Skills: GoRuby
An Hour Ago
Easy Apply
Remote
29 Locations
Easy Apply
Entry level
Entry level
Cloud • Security • Software • Cybersecurity • Automation
As an Intermediate Site Reliability Engineer in FinOps at GitLab, you'll ensure systems are scalable, reliable, and financially optimized. Your role involves automating cost management, collaborating with finance and engineering teams, and promoting FinOps principles across operations for cost optimization and financial accountability.
Top Skills: AnsibleAWSGCPTerraform

What you need to know about the Pune Tech Scene

Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account