Machine Learning Engineer
Cognite - AI for Industry17 days ago
Bengaluru, IndiaMid Level / Senior
H1B Sponsor
Responsibilities
- Build, test, and maintain robust data pipelines for large-scale industrial datasets.
- Assist in deploying data science prototypes to production.
- Write high-quality, testable Python code, primarily creating RESTful/gRPC APIs for ML capabilities.
- Containerize applications (Docker) and navigate Linux to handle massive volumes of unstructured industrial data.
- Run experiments, fine-tune existing foundational models, and evaluate ML libraries to enhance document parsing and entity matching.
- Write rigorous unit/integration tests and monitor deployed models for drift and performance.
- Collaborate closely with senior staff to deliver features on the technical roadmap.
Requirements
- Bachelor’s or Master's degree in Computer Science, Data Science, Software Engineering, or a related field.
- 3–6 years of industry experience in software engineering with a strong focus on machine learning.
- Strong programming skills in Python, with a proven ability to write clean, modular, and testable code.
- Solid hands-on experience with frameworks like PyTorch, TensorFlow, Hugging Face, or LangChain.
- Experience working with large datasets and writing complex SQL queries.
- Familiarity with MLOps basics, containerization (Docker), Linux command-line tools, and cloud services.
- Solid understanding of core computer science fundamentals, data structures, and algorithms.