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.
The Role
As an AI / ML Engineer on the ATLAS AI Co-Innovation team, you will build and optimize AI models and agent workflows that bring industrial data to life. Working closely with product teams, engineers, and customers, you’ll turn cutting-edge GenAI research into scalable, real-world solutions that drive smarter industrial operations. This role is perfect for someone who thrives at the intersection of innovation, problem-solving, and hands-on engineering.
Responsibilities
- 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
What We Are Looking For
Extensive software engineering experience in building high performance, large-scale ML systems (Preferably 3+ 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? \uD83C\uDFC6 \uD83D\uDE80
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 sustainability for clients and allows front-line workers, as well as domain experts, to make better decisions every single day.Cognite Earns 2023 Microsoft Partner of the Year Award; Recognized as a Global Leader in Energy & Resources and Industrials & ManufacturingFrost & Sullivan named Cognite a Technology Innovation Leader!Built In 2024 Best Places to Work in Austin, TX and Houston, TXCognite Recognized as 2024 Microsoft Energy and Resources Partner of the YearMost recently Cognite Data Fusion® Achieved Industry First DNV Compliance for Digital Twins \uD83E\uDD47A snapshot of our many perks and benefits as a Cogniter
Competitive compensation401(k) with employer matchingCompetitive health, dental, vision & disability coverages for employees and all dependentsUnlimited PTOPaid Parental Leave ProgramEmployee Referral ProgramJoin a team of 60+ different nationalities \uD83C\uDF10 with Diversity, Equality and Inclusion (DEI) in focus \uD83E\uDD1D.A highly modern and fun working environment with sublime culture across the organization, follow us on Instagram @cognitedata \uD83D\uDCF7 to know moreOpportunity to work with and learn from some of the best people on some of the most ambitious projects found anywhere, across industriesJoin our HUB \uD83D\uDDE3️ to be part of the conversation directly with Cogniters and our partners.Paid mobile phone and WiFIAll 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.