Senior AI/ML Engineer

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 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. 

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

• End-to-End Prototyping Build cross-stack prototypes using ATLAS AI, CDF, and open-source AI frameworks to solve real customer challenges.
• Agent Workflow Design Design and implement multi-agent workflows that combine LLMs, tool use, and reasoning over industrial data.
• Tech Exploration & Integration Evaluate and integrate new GenAI tools, open-source frameworks, and APIs into ATLAS AI workflows.
• System Optimization Benchmark performance, tune retrieval and reasoning pipelines, and ensure scalability in real-world industrial deployments.
• Collaboration & Co-Innovation Work with solution engineers and customer teams to align models and agent behaviors with business value and industrial constraints.

What We're Looking For - Must-Have Skills

• 5+ years of experience in AI/ML engineering, with hands-on delivery of models.
• Proficiency in working with foundation models (LLMs), including :Prompt engineering, evaluation, and (when relevant) fine-tuning.
• RAG pipelines and integration with knowledge bases or vector databases.
• Strong Python skills with experience using frameworks such as LangChain, Transformers, or similar.
• Understanding of cloud-native development, model training workflows, and ML pipeline orchestration (e.g., data labeling, feature selection, model retraining).
• Proven ability to write clean, maintainable, and scalable code, following engineering best practices for testing, version control, and review.
• A maker mindset with bias toward rapid iteration, showing rather than telling, and learning by doing.

Bonus Skills

• Experience with Cognite Data Fusion (CDF).
• Experience integrating AI workflows with time series, asset hierarchies, or knowledge graphs.
• Deep learning or traditional ML background (e.g., model architecture selection, hyperparameter tuning, evaluation pipelines).
• Understanding of industrial data types (e.g., time series, contextual events, industrial knowledge graphs).
• Experience labeling industrial datasets, including annotation strategies and working with imperfect or sparse labels.

 Join the global Cognite community - Join an organization of 70 different nationalities with Diversity, Equality and Inclusion (DEI) in focus - Office location Rathi Legacy (Rohan Tech Park ) Hoodi (Bengaluru) - A highly modern and fun working environment with sublime culture across the organization, follow us on Instagram @cognitedata  to know more - Flat structure with direct access to decision-makers, with minimal amount of bureaucracy - 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. - Hybrid work environment globally
Why choose Cognite? Join us in making a real and lasting impact in one of the most exciting and fastest-growing new software companies in the world. We have repeatedly demonstrated that digital transformation, when anchored on strong DataOps, drives business value and sustainabilityfor clients and allows front-line workers, as well as domain experts, to make better decisions every single day. We were recognized as one of CNBC's top global enterprise technology startups powering digital transformation And just recently, Frost & Sullivan named Cognite a Technology Innovation Leader Most recently Cognite Data Fusion Achieved Industry First DNV Compliance for Digital Twins
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.

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.

---