We are looking for a Forward Deployed ML Engineer, with an interest in solving difficult real-world problems to join our Core AI team.
This role will require extensive working with custom architectures on PyTorch. Previous experience working with PyTorch on complex Convolutional Neural Network, Graph Neural Network, and/or Transformers is expected. This role is equivalent to the Delta team Forward Deployed Engineer role at Palantir that we would recommend you read up on.
This is a remote-first position, with a preference for applicants based in Canada.
StratumAI & the Core AI team
StratumAI creates advanced resource models that empower operating mining companies to make smarter, more efficient decisions about mineral extraction. Using our proprietary adaption of neural networks, we produce 3D maps that highlight where and how to target drilling, blasting, and development activities for optimal outcomes. Our long-term vision is to build a massive AI engine capable of making every decision in a mining operation, down to moving individual rocks. We are looking for engineers to help make this vision a reality.
The Core AI team responsibilities include:
- Conducting foundational research to design and implement more accurate AI resource modelling techniques that we can use for many different mining operations.
- Applying and refining existing architecture to train resource models for a specific mine.
- Communicating the quality, metrics, performance, and methodology of our models to non-ML technical external stakeholders.
- Identifying new opportunities to improve mining operations (such as modelling other parameters) for a given client and getting their support in doing so.
- Tracking model performance of deployed models over time and identifying ways to make existing models better.
Role Requirements
- 2+ years of industry machine learning experience.
- Excellent proficiency in implementation of custom neural network architectures in Python (Pytorch).
- Strong foundational knowledge of deep learning with an emphasis on transformers, convolutional neural networks, and/or graph neural networks.
- Adept at working with a high degree of autonomy and initiative to solve complex problems.
- Strong technical communication skills (oral + visual).
- You will be responsible for conveying to other non-machine learning engineers in the mining industry why our models are good. You will work with geologists and business development roles to tell the story.
Nice to Have’s
- An open motor vehicle driving license.
- Ability to speak another language with high proficiency.
Responsibilities
- Adapt Stratum’s deep learning models to a given mine.
- Develop and maintain high-quality machine learning code using Python.
- Unlock ways to create better resource and metallurgical models with deep learning for one specific mine, or sometimes mines in general
- 60% of your time is dedicated to applying ML to one specific mine (applied ML), 40% is dedicated to applying ML to mines in general (foundational ML).
- Take part in at least 2 mine visits a year and engage with our clients on the ground
- Over time, grow into a senior engineer who can identify additional ways our technology can be leveraged with our existing clientele
Culture Fit
We are looking for people who are:
- Self-driven: You thrive on autonomy and enjoy solving complex problems to get the most impactful solution.
- Detail-oriented: You can identify the one line that differentiates good code from a bad one.
- Curious about real-world problems: You have interest in learning how industries outside of your professional domain operate.
- Flexible with time: You’re open to working not long but occasionally odd hours to accommodate overseas clients (some weekend calls may be required).
Our Tech Stack
- AI Stack: Python & Pytorch
- Backend: Python (FastAPI) + Pydantic + DynamoDB
- Infrastructure: AWS, Azure
Our long-term vision is to build a massive AI engine capable of making every decision in a mining operation, down to moving individual rocks. If you’re an exceptional engineer interested to helping make this vision a reality we look forward to reviewing your application and working together.