Machine Learning Engineer - LLM & Generative AI

What Youll Do

• 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

Who You Are

• Extensive software engineering experience in building high performance, large-scale ML systems (Preferably 3 to 5 years of industry experience in Oil Gas, manufacturing etc).
• 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.
• B.E / BTech 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.
• Experience with distributed parallel training, large-scale multi-modal foundation and generative models strongly preferred.
• Experience with RAG, dual encoders, or related fields 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 JAX.
• Excellent problem-solving and troubleshooting abilities, along with a strong analytical mindset and persistence in resolving problems.
• Strong communication skills and the ability to work effectively in a collaborative research environment.
• A track record of publications and contributions to the machine learning and natural language processing communities strongly preferred.

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 sustainability for clients and allows front-line workers, as well as domain experts, to make better decisions every single day. We were recognized as one of CNBCs 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

A snapshot of our many perks and benefits as a Cogniter

• Competitive salary and benefits, including a pension plan, insurance, parental benefits, and much more.
• Join an organization of 70 different nationalities with Diversity, Equality and Inclusion (DEI) in focus
• 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.
• Gain perks like a paid mobile telephone subscription and broadband connection.
• Stay fueled between meals with snacks and drinks on the house.

Apply today!

If youre 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. Please do not hesitate to contact our Talent Acquisition team with any questions - Smriti Joshi at smriti.joshi@cognite.com.

We encourage you to follow us on Cognite LinkedIn ; we post all our openings there.

Join the global Cognite Community!

- Join an organization of 70 different nationalities with Diversity, Equality and Inclusion (DEI) in focus

- ITPL office location (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 sustainability for clients and allows front-line workers, as well as domain experts, to make better decisions every single day. We were recognized as one of CNBCs 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 youre 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. Please do not hesitate to contact our Talent Acquisition team with any questions -

We encourage you to follow us on Cognite LinkedIn ; we post all our openings there.