Machine Learning Developer


Here at StratumAI, we work towards improving the mining industry using ML to make better resource models, reduce waste, and improve efficiency. We are looking for an adaptable and capable Machine Learning Developer to join our startup’s technical team and help us move towards changing the global mining industry.

You’ll be joining a diverse team of engineers and data scientists to work on building, improving, and maintaining end-to-end machine learning systems. If you are interested in working on ML problems in industry while also tackling fundamental machine learning research, then this is the position for you. To do this job successfully, you need exceptional skills in machine learning and data science with a healthy dose of curiosity to learn new things on the fly. Interest in working at a growing startup is a must!


  • Developing deep learning models to solve novel problems for mines around the world
  • Doing data analysis and data science to extract insights from mining data
  • Extracting technical requirements from client meetings and needs
  • Conducting applied research to improve model performance for a given use case
  • Applying existing ML libraries and frameworks for tasks such as model and data uncertainty
  • Researching and implementing new models, diagnostic methods, and tools that assist meeting common client requirements
  • Keep abreast of developments in the field


  • Experience working with machine learning in computer vision, NLP, recommender systems, and scientific applications
  • Strong background in probability, machine learning, and data science
  • Understanding of data structures, data modeling and software architecture
  • Ability to write robust code in Python
  • Experience with ML frameworks/libraries (Tensorflow, PyTorch, Jax, sk-learn)
  • Familiarity with Git, experiment tracking tools (WandB, Comet, etc.)
  • Strong experience with data analysis/processing libraries such as pandas and numpy
  • Excellent communication skills for both technical and non-technical audiences
  • Self-learner and motivated to pick up new skills
  • Outstanding analytical and problem-solving skills

Nice to Have

  • Strong software engineering skills
  • Previous experience working at startups
  • Publications related to machine learning, deep learning, robotics and/or statistics
  • Oral/written fluency in a language besides English

To apply, please email your CV to

<< Back to Careers