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