Machine Learning Ops Engineer

The role

You will build and maintain a platform to train, evaluate, and serve AI models for the mining industry. You will work on the Infrastructure Team to create robust infrastructure, tooling, and production-grade QA/QC systems.

What you will do

• Develop robust and well-tested code for core internal tools
• Build and maintain a custom end-to-end MLOps platform
• Develop production-grade QA/QC systems for deployed AI models
• Create specialized tools for mining data
• Refactor and productionize code created by the client services team
• Provide technical expertise to the client services team
• Manage infrastructure for data processing, model training, and serving
• Mentor junior engineers, perform code reviews, and write documentation
• Proactively identify technical challenges and drive improvement initiatives

Required qualifications

• Bachelor's degree in Computer Science, Engineering, or related fields OR equivalent experience in software development and ML engineering
• 3+ years of industry experience
• Kubernetes, PyTorch
• Advanced Python programming skills
• Proficiency with data science libraries (numpy, pandas)
• Experience with visualization tools
• Ability to write modular, robust, and tested Python code
• Strong debugging skills for complex ML systems
• Implementation of neural network models and training workflows
• Understanding of model architecture selection
• Knowledge of model evaluation techniques
• Creating experiment tracking systems

Nice to have

• Previous experience working at startups
• Familiarity with Git, experiment tracking tools (WandB, Comet, etc.)
• Experience working on production machine learning using tools such as KubeFlow, MLFlow, AirFlow, Seldon Core, DVC, Spark, etc.
• Written/oral fluency in a language besides English
• Experience optimizing data processing pipelines and/or neural network models
• Proficiency in a lower-level programming language or GPU programming
• Experience with data application frameworks
• Full stack development experience
• Experience with experiment tracking systems and ML model monitoring
• Background in mining or resource modeling

About the company

Stratum is a mining software company with machine learning models as its core product. The company uses 3D maps to predict mineral distribution using proprietary ML protocols.