Machine Learning Engineer, AI Powered: Custom Models

GitLab

GitLab

Software Engineering, Data Science
Remote
Posted on Jul 8, 2024

The GitLab DevSecOps platform empowers 100,000+ organizations to deliver software faster and more efficiently. We are one of the world’s largest all-remote companies with 2,000+ team members and values that foster a culture where people embrace the belief that everyone can contribute. Learn more about Life at GitLab.

An overview of this role

The Custom Models team 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. They will work collaboratively with numerous teams to ensure a complete lifecycle of assessing models, fine-tuning models, evaluating models, storing models, deploying models, implementing models as underlying engines behind AI agents, and protecting these models through means of various hosting techniques.

Why us? This isn't just a job; it's your chance to shape the future of AI at GitLab. Your expertise in backend development will be critical to your success. Ready to dive into the future of AI at GitLab? Apply now! We're excited to meet potential candidates like you and welcome a new star to our team. Let's shape the future together!

What you’ll do

  • Develop improvements to models to generate new content using machine learning models in a secure, well-tested, and performant way.
  • Work with highly complex data for feature development using machine learning models.
  • Collaborate with product managers, engineers, and other stakeholders as a machine learning specialist.
  • Advocate for improvements to product quality, security, and performance.
  • Solve technical problems of moderate scope and complexity.
  • Craft code that meets our internal standards for style, maintainability, and best practices for a high-scale machine-learning environment. Maintain and advocate for these standards through code review.
  • Confidently ship small features and improvements with minimal guidance and support from other team members. Collaborate with the team on larger projects.
  • Participate as a reviewer or project maintainer in one or more engineering projects.
  • Participate in Tier 2 or Tier 3 weekday, weekend, and occasional night on-call rotations to assist with troubleshooting product operations, security operations, and urgent engineering issues.

What you’ll bring

  • A relevant Master’s degree and 2 or more years of experience in ML or PhD degree with a focus on Machine Learning or Data Science.
  • Professional experience with Python
  • Experience with performance and optimization problems and a demonstrated ability to both diagnose and prevent these problems
  • Comfort working in a highly agile, intensely iterative software development process
  • Demonstrated ability to onboard and integrate with an organization long-term
  • Positive and solution-oriented mindset
  • Effective communication skills: Regularly achieve consensus with peers, and clear status updates
  • An inclination towards communication, inclusion, and visibility
  • Experience owning a project from concept to production, including proposal, discussion, and execution.
  • Self-motivated and self-managing, with strong organizational skills.
  • Demonstrated ability to work closely with other parts of the organization
  • Share our Values, and work in accordance with those values
  • Ability to thrive in a fully remote organization

Two of more of

  • Professional experience with prompt engineering and Retrieval Augmented Generation (RAG)
  • Experience building, training, and implementing deep learning models.
  • Experience with a deep learning framework such as PyTorch or TensorFlow
  • Professional experience fine-tuning LLMs
  • Design, construction or operation of MLOps infrastructure

Bonus qualifications

  • Have contributed a Merge Request to GitLab
  • Have contributed to ML open source projects

About the team

The Custom Models team is a new team formed with internal GitLab team members from around the globe. Engineers are primarily located across various European countries, with some distributions in Australia, New Zealand, America, and Canada. This team is growing, and there are currently 7 positions available.

The team works closely with these other teams within the organization:

  • AI Framework

  • MLOps

  • Model Validation

  • Duo Chat

How GitLab will support you

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.

The base salary range for this role’s listed level is currently for residents of listed locations only. Grade level and salary ranges are determined through interviews and a review of education, experience, knowledge, skills, abilities of the applicant, equity with other team members, and alignment with market data. See more information on our benefits and equity. Sales roles are also eligible for incentive pay targeted at up to 100% of the offered base salary.

California/Colorado/Hawaii/New Jersey/New York/Washington/DC pay range
$112,000$240,000 USD

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.