AI / ML Engineer

Job Description

Job Description

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

Cognite Product Tour 2025

Cognite Product Tour 2024

Cognite Product Tour 2023

Data Contextualization Masterclass 2023

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.

Our core product, Cognite Data Fusion (CDF), is a leading industrial DataOps platform. Building on CDF, ATLAS AI is our innovative offering designed to deliver bold, customer-centric innovation powered by artificial intelligence.

The Role

As an AI / ML Engineer in the ATLAS AI Co-Innovation team, you will help push the technical boundaries of what’s possible with industrial GenAI. You’ll design and optimize advanced AI models and agent architectures that interact with complex, real-world industrial data. You’ll operate at the technical core of customer-facing coinnovation, working closely with solution engineers, product teams, and customer data to build smart, scalable AI components that power next-generation industrial workflows.

This role demands strong AI / ML engineering skills, deep curiosity, and the ability to adapt cutting-edge research into usable, high-impact solutions.

Responsibilities

• Design and apply foundation models to domain-specific tasks with a focus on prompt engineering, reasoning workflows, tool use, and real-world accuracy
• Build modular, production-ready agent workflows integrated with CDF and ATLAS AI using tools, memory, reasoning chains, and APIs
• Evaluate and integrate new GenAI tools, open-source frameworks, and APIs into ATLAS AI workflows
• Benchmark model performance, tune retrieval and reasoning pipelines, and ensure scalability in industrial deployments
• Collaborate with solution engineers and customer teams to align AI models and agent behaviors with business value and industrial needs

What We Are Looking For

Bonus Skills

Why choose Cognite? \uD83C\uDFC6 \uD83D\uDE80

A snapshot of our many perks and benefits as a Cogniter

All candidates must be legally authorized to work in the United States without the need for current or future company sponsorship for employment visa status.

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.

We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.