Senior Machine Learning Engineer, ATLAS AI

Position Overview

Cognite is a global SaaS forerunner leveraging AI and data to solve complex industrial challenges. As an AI/ML Engineer on the ATLAS AI Co-Innovation team, you will build and optimize AI models and agent workflows that transform industrial operations. Join an award‐winning organization recognized for its innovation and impact across Oil & Gas, Chemicals, Pharma, Manufacturing, and Energy sectors.

Key Responsibilities

• Collaborate with researchers and engineers to design, develop, and refine large language and generative models for various applications.
• Utilize 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 in language modeling, generative modeling, and machine learning.
• Deliver the next generation of innovation in trustworthy Industrial Data Insights & Collaboration.

Required Qualifications

• Extensive software engineering experience in building high performance, large-scale ML systems (preferably 3+ years in industry).
• Strong communication skills with the ability to work effectively in a collaborative research environment.
• Demonstrated leadership abilities with experience guiding and mentoring a team of ML engineers.
• Excellent problem-solving and troubleshooting skills combined with a strong analytical mindset.
• Deep theoretical and practical background in NLP with hands-on experience in state-of-the-art LLM architectures.
• An advanced degree (ideally a Ph.D.) in Computer Science, Electrical Engineering, or a related field with specialized research in machine learning, NLP, or generative AI techniques.
• Proficiency in deep learning frameworks (e.g., PyTorch, TensorFlow) and associated NLP libraries.
• Solid programming skills in Python with the ability to write clean, efficient, and well-documented code.
• At least 3 years of experience working in complex, matrixed organizations with cross-functional projects.
• Familiarity with cloud-based infrastructure and experience in deploying large-scale ML models in production environments.
• Experience with JAX.

Preferred Qualifications

• Experience with distributed parallel training, large-scale multi-modal foundation and generative models strongly preferred.
• Expertise in LLM tuning including prompt engineering, instruction tuning, RLHF, parameter-efficient tuning, and fine-tuning strongly preferred.
• Familiarity with training multi-modal foundation models and relevant frameworks strongly preferred.
• Experience with RAG, dual encoders, or related fields strongly preferred.
• A track record of publications and contributions to machine learning and NLP communities strongly preferred.

Benefits & Perks

• Competitive compensation
• 401(k) with employer matching
• Competitive health, dental, vision & disability coverages for employees and all dependents
• Unlimited PTO
• Paid Parental Leave Program
• Employee Stock Purchase Program (ESPP)
• Employee Referral Program
• Be part of a diverse team representing 60+ nationalities with a strong focus on Diversity, Equality, and Inclusion (DEI)
• A modern, fun working environment with a vibrant culture – follow us on Instagram @cognitedata
• Opportunities to work with and learn from top professionals on ambitious projects across industries
• Access to our HUB for direct conversations with Cognite and our partners
• Paid mobile phone and WiFi