Senior Machine Learning Engineer
Cognite - AI for Industry18 days ago
Responsibilities
- Design, train, test, and deploy ML models for document parsing and entity matching.
- Write high-quality, scalable production code in Python and wrap ML models into robust APIs.
- Implement and maintain CI/CD pipelines for ML models and manage containerized applications.
- Translate high-level system architectures into scalable data pipelines and microservices.
- Design automated testing for ML pipelines and monitor deployed models for performance.
- Collaborate with product managers to define ML capabilities and provide technical mentorship.
Requirements
- Bachelor’s or Master's degree in Computer Science, Data Science, Software Engineering, or a related field.
- 6–10 years of industry experience in software engineering with a focus on machine learning and MLOps.
- Strong programming skills in Python and experience with backend web frameworks.
- Expertise in frameworks like PyTorch, TensorFlow, or Hugging Face Transformers.
- Hands-on experience with containerization, CI/CD tools, and deploying models on cloud platforms.
- Solid understanding of software architecture, data structures, and algorithms.
Tech Stack
Apache AirflowApache SparkAWSAzureDatabricksDjangoDockerFastAPIFlaskGitHub ActionsGoogle Cloud PlatformHugging Face TransformersJenkinsKubernetesLinuxPythonPyTorchTensorFlow