Machine Learning Advocate Engineer

Intern / Student, Full-time · Maria01 (Helsinki)

About this role
We’re a close-knit, ambitious team of 30, driven by a mission to make this world a better place.  We want you on our team if you’re ready to lead with heart and focus.

We are looking for a Machine Learning Advocate Engineer to work on benchmarking and testing the latest machine learning models on different GPU hardware, analyzing the latest model architectures, and writing down the findings in accessible blog articles, as well as participating as a ML engineer working on model deployment for our inference customers.

Candidates preferred for this role should be currently enrolled in a PhD program, or in the late stages of their master’s studies.
What you will do
  • Develop, run and maintain AI model benchmarks for different workloads and GPU types.
  • Review and analyse the latest models and model optimizations in image generation, language modelling and other domains.
  • Write high-quality articles, tutorials, and case studies related to model performance, architecture, and effective inference deployment.
  • Work closely with machine learning and software engineering teams to contribute the latest model performance optimizations to open source projects.
  • Create and maintain tooling and services: both internal facing (internal research) and external facing (open-source libraries).
  • Implement and optimize open-source libraries for performance and accuracy, ensuring production readiness and employing cutting-edge technology and innovative approaches.
  • Collaborate with the open-source community and scout for novel inference & training frameworks as well as research approaches.
  • Complete other tasks as necessary to contribute to the progress of the company.
For this role you will need
  • Shown progress towards PhD degree, or exceptional knowledge in late stages of master’s degree in the fields of Computer Science, Machine Learning, or relevant technical discipline. 
  • Excellent writing and communication skills in English.
  • Ability to explain complex technical concepts in a clear and engaging manner.
  • Good grasp of the underlying mathematical principles.
  • Experience contributing to popular open-source libraries such as PyTorch, Tensorflow, JAX, vLLM. 
  • Proficiency in frameworks such as PyTorch, Jax or TensorFlow.
  • High attention to detail. 
  • Willingness to learn and adapt to the constant change in the machine learning ecosystem.
What we offer
  • Company equity - you’re in this with us!
  • Competitive salary and benefits, including health insurance, lunch benefit, annual budget to spend as you wish (i.e. sport, transport, wellness, culture)
  • Flexible work environment
  • Opportunity to work with cutting-edge AI
  • Growth in a mission-driven company
About us
DataCrunch is a dynamic and innovative company dedicated to making this world better. At DataCrunch, our team plays a key role in the decision-making process. If you're passionate about innovation and enjoy experimenting with cutting-edge technologies, DataCrunch is the perfect fit. Join our team of diverse individuals and be part of creating something extraordinary.
We are looking forward to hearing from you!
Thank you for your interest in DataCrunch. Please fill out the following short form. Should you have difficulties with the upload of your data, please send an email to lena@datacrunch.io
Uploading document. Please wait.
Please add all mandatory information with a * to send your application.