Senior Machine Learning Engineer โ€“ LLM & Generative AI

undefined ago โ€ข 3-5 Years โ€ข Research Development

Job Summary

Job Description

Cognite is seeking a Senior Machine Learning Engineer to join its AI team, focusing on LLM and Generative AI. The role involves leveraging AI and data to solve complex industrial challenges using platforms like Cognite Atlas AI and Cognite Data Fusion. This position offers a unique opportunity to work on state-of-the-art language models, contribute to transformative research, and make a significant impact on the future of Industrial DataOps, collaborating with diverse teams to derive insights from complex industrial data.
Must have:
  • Collaborate with researchers and engineers to design, develop, and refine large language models and generative models.
  • Utilize expertise in machine learning and natural language processing to develop novel algorithms and methodologies.
  • Implement, train, and fine-tune LLM and GPT-like models on large-scale datasets.
  • Stay up to date with the latest research advancements and techniques in language modeling and generative AI.
  • Deliver innovation in trustworthy Industrial Data Insights & Collaboration.
  • Extensive software engineering experience in building high performance, large-scale ML systems (3 to 5 years).
  • Strong communication skills and ability to work effectively in a collaborative research environment.
  • Demonstrated leadership abilities, with experience in guiding and mentoring ML engineers.
  • Excellent problem-solving and troubleshooting abilities with a strong analytical mindset.
  • Strong theoretical and practical background in NLP, including state-of-the-art LLM architectures.
  • Advanced degree (ideally Ph.D.) in Computer Science, Electrical Engineering, or related field with specialized research.
  • Proficiency in deep learning frameworks (e.g., PyTorch, TensorFlow) and solid Python programming skills.
  • 3 years of experience working in a complex, matrixed organization.
  • Familiarity with cloud-based infrastructure and experience deploying large-scale machine learning models in production.
Good to have:
  • Experience with distributed parallel training, large-scale multi-modal foundation and generative models.
  • Collaborate on modeling techniques for LLM tuning (prompt engineering, instruction tuning, RLHF, parameter-efficient tuning, fine-tuning).
  • Familiarity with training multi-modal foundation models and relevant frameworks.
  • Experience with RAG, dual encoders, or related fields.
  • A track record of publications and contributions to the machine learning and natural language processing communities.
Perks:
  • Competitive Compensation including base plus bonus
  • 401(k) with 4% employer matching
  • Health, Dental, Vision & Disability Coverages with premiums fully covered for employees and all dependents
  • Unlimited PTO + flexibility to enjoy it
  • 18 Company Holidays including the week between Christmas & New Years
  • Paid Parental Leave Program
  • Employee Stock Purchase Program (ESPP)
  • Employee Referral Program
  • In office catered lunches 3x per week via Colorfull Catering + Fully Stocked Fridges in the offices
  • Join a team of 70 different nationalities with Diversity, Equality and Inclusion (DEI) in focus
  • A highly modern and fun working environment with sublime culture
  • Opportunity to work with and learn from some of the best people on some of the most ambitious projects
  • Join our HUB to be part of the conversation directly with Cogniters and our partners
  • Paid mobile phone and WiFI
  • Chance to meet Spot (dog)

Job Details

About Cognite

Embark on a transformative journey with Cognite, a global SaaS forerunner in leveraging AI and data to unravel complex business challenges through our cutting-edge offerings including Cognite Atlas AI, an industrial agent workbench, and the Cognite Data Fusion (CDF) platform. We were awarded the 2022 Technology Innovation Leader for Global Digital Industrial Platforms & Cognite was recognized as 2024 Microsoft Energy and Resources Partner of the Year. In the realm of industrial digital transformation, we stand at the forefront, reshaping the future of Oil & Gas, Chemicals, Pharma and other Manufacturing and Energy sectors. Join us in this venture where AI and data meet ingenuity, and together, we forge the path to a smarter, more connected industrial future.

Learn more about Cognite here

Our values

Impact: Cogniters strive to make an impact in all that they do. We are result-oriented, always asking ourselves.

Ownership: Cogniters embrace a culture of ownership. We go beyond our comfort zones to contribute to the greater good, fostering inclusivity and sharing responsibilities for challenges and success.

Relentless: Cogniters are relentless in their pursuit of innovation. We are determined and deliverable (never ruthless or reckless), facing challenges head-on and viewing setbacks as opportunities for growth.

The rapid advances of Artificial intelligence (AI), especially around Generative AI, and in particular LLMs like OpenAI GPT family of models, have driven innovations starting to disrupt existing market spaces across industry. Many businesses are betting on AI as the next growth opportunity.

Who you are:

We are looking for a highly motivated and talented Senior Machine Learning Engineer to join our AI team. The ideal candidate will have experience in Large Language Modeling (LLM) and DataOps. We are seeking problem solvers, passionate and collaborative teammates, and those willing to roll up their sleeves while making a difference. If you are interested in the opportunity to pursue purposeful work, join a mission-driven team, and build a rewarding career while having fun, then this is the perfect job for you!

Opportunity:

Machine learning experts on our team can utilize applied science and software development skills to help tweak Foundation Models that help us address some of the hardest problems towards our vision in making insights from complex Industrial data simple & easy to access. You will work in an exciting and fast-paced environment, collaborating closely with multiple teams across the company. You will work as part of an organization that brings together talent from diverse backgrounds including software engineering, big data, machine learning, AI, and Industrial data architects making our team an exciting place to work. We value and encourage diversity in the belief that our differences make us and our products better.

This position offers a unique chance to work on state-of-the-art language models and contribute to transformative research with the vision of saving lives with Industrial Data. You will be part of a dynamic team of researchers and engineers who are passionate about pushing the boundaries of machine learning and natural language understanding in the Industrial domain. Join us and make a significant impact on the future of Industrial DataOps.

What You'll Do

  • Collaborate with researchers and engineers to design, develop, and refine large language models and generative models for various applications.
  • Utilize your expertise in machine learning and natural language processing to develop novel algorithms and methodologies for generative modeling tasks.
  • Implement, train, and fine-tune LLM and GPT-like models on large-scale datasets to ensure optimal performance and accuracy.
  • Stay up to date with the latest research advancements and techniques in the field of language modeling, generative modeling, and machine learning.
  • Deliver the next generation of innovation in trustworthy Industrial Data Insights & Collaboration

Who You Are

  • Extensive software engineering experience in building high performance, large-scale ML systems (Preferably 3 to 5 years of industry experience in the relevant fields).
  • Strong communication skills and the ability to work effectively in a collaborative research environment.
  • Demonstrated leadership abilities, with experience in guiding and mentoring a team of ML engineers.
  • Excellent problem-solving and troubleshooting abilities, along with a strong analytical mindset and persistence in resolving problems.
  • Strong theoretical and practical background in NLP including experience with state-of-the-art LLM architectures optimized for handling large context and low inference cost.
  • Advanced degree, ideally a Ph.D., in Computer Science, Electrical Engineering, or a related field with specialized research in areas such as machine learning, NLP, Large Language Models (LLMs), multi-modal foundation models, and generative AI techniques.
  • Proficiency in deep learning frameworks (e.g., PyTorch, TensorFlow, etc.) and libraries commonly used in NLP and Generative AI.
  • Solid programming skills in Python and the ability to write clean, efficient, and well-documented code.
  • 3 years of experience working in a complex, matrixed organization involving cross-functional or cross-business projects.
  • Experience with distributed parallel training, large-scale multi-modal foundation and generative models strongly preferred.
  • Collaborate on modeling techniques to support the full spectrum of LLM tuning including prompt engineering, instruction tuning, Reinforcement Learning from Human Feedback (RLHF), parameter-efficient tuning, and fine-tuning strongly preferred.
  • Familiarity with training multi-modal foundation models and relevant frameworks strongly preferred.
  • Familiarity with cloud-based infrastructure and experience deploying large-scale machine learning models in production environments.
  • Experience with RAG, dual encoders, or related fields strongly preferred.
  • Experience with JAX.
  • A track record of publications and contributions to the machine learning and natural language processing communities strongly preferred.

Why choose Cognite? ๐Ÿ† ๐Ÿš€

A snapshot of our many perks and benefits as a Cogniter

  • Competitive Compensation including base plus bonus
  • 401(k) with 4% employer matching
  • Health, Dental, Vision & Disability Coverages with premiums fully covered for employees and all dependents
  • Unlimited PTO + flexibility to enjoy it
  • 18 Company Holidays including the week between Christmas & New Years
  • Paid Parental Leave Program
  • Employee Stock Purchase Program (ESPP)
  • Employee Referral Program
  • Company Paid Friday Lunch via DoorDash + Fully Stocked Fridges in the offices
  • Join a team of 70 different nationalities ๐ŸŒ with Diversity, Equality and Inclusion (DEI) in focus ๐Ÿค.
  • A highly modern and fun working environment with sublime culture across the organization, follow us on Instagram @cognitedata ๐Ÿ“ท to know more
  • Opportunity to work with and learn from some of the best people on some of the most ambitious projects found anywhere, across industries
  • Join our HUB ๐Ÿ—ฃ๏ธ to be part of the conversation directly with Cogniters and our partners.
  • Paid mobile phone and WiFI
  • A pet lover? Get the chance to meet Spot ๐Ÿถ!

Donโ€™t meet every single requirement? Studies have shown that women and people of colour are less likely to apply to jobs unless they meet every single qualification. We are dedicated to building a diverse, inclusive and authentic workplace, so if youโ€™re excited about this role but your past experience doesnโ€™t align perfectly with every qualification in the job description, we encourage you to apply anyways. You may be just the right candidate for this or other roles.

A snapshot of our many perks and benefits as a Cogniter

  • Competitive Compensation including base plus bonus
  • 401(k) with 4% employer matching
  • Health, Dental, Vision & Disability Coverages with premiums fully covered for employees and all dependents
  • Unlimited PTO + flexibility to enjoy it
  • 18 Company Holidays including the week between Christmas & New Years
  • Paid Parental Leave Program
  • Employee Stock Purchase Program (ESPP)
  • Employee Referral Program
  • In office catered lunches 3x per week via Colorfull Catering + Fully Stocked Fridges in the offices
  • Join a team of 70 different nationalities ๐ŸŒ with Diversity, Equality and Inclusion (DEI) in focus ๐Ÿค.
  • A highly modern and fun working environment with sublime culture across the organization, follow us on Instagram @cognitedata ๐Ÿ“ท to know more
  • Opportunity to work with and learn from some of the best people on some of the most ambitious projects found anywhere, across industries
  • Join our HUB ๐Ÿ—ฃ๏ธ to be part of the conversation directly with Cogniters and our partners.
  • Paid mobile phone and WiFI
  • A pet lover? Get the chance to meet Spot ๐Ÿถ!

Why choose Cognite? ๐Ÿ† ๐Ÿš€

Apply today!

If you're excited about the opportunity to work at Cognite and make a difference in the tech industry, we encourage you to apply today! We welcome candidates of all backgrounds and identities to join our team.

We encourage you to follow us on Cognite LinkedIn; we post all our openings there.

Equal Opportunity

Cognite is committed to creating a diverse and inclusive environment at work and is proud to be an equal opportunity employer. All qualified applicants will receive the same level of consideration for employment; everyone we hire will receive the same level of consideration for training, compensation, and promotion.

We ask for gender as part of our application because we want to ensure equal assessment in the recruitment process. Your answer will help us reach this commitment! However, the question about gender is optional and your choice not to answer will not affect the assessment of your application in any way.

Similar Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Similar Skill Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Jobs in Phoenix, Arizona, United States

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Research Development Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

About The Company

Phoenix, Arizona, United States (Hybrid)

Pau, Nouvelle-Aquitaine, France (Hybrid)

Phoenix, Arizona, United States (Hybrid)

Bengaluru, Karnataka, India (Hybrid)

Bengaluru, Karnataka, India (Hybrid)

Oslo, Oslo, Norway (Hybrid)

Oslo, Oslo, Norway (Hybrid)

Oslo, Oslo, Norway (Hybrid)

Phoenix, Arizona, United States (Hybrid)

Phoenix, Arizona, United States (Hybrid)

View All Jobs

Get notified when new jobs are added by Cognite

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug