Senior Machine Learning Engineer

Zagreb, Croatia

With over 300 million downloads globally, Photomath is  the #1 app in the world to learn math and one of the most popular educational apps of all time.

Powered by advanced machine learning technology, the app instantly scans, accurately solves, and intuitively explains printed and handwritten math problems to users through step-by-step explanations.

Since its launch in 2014, our award-winning app has topped App Store & Google Play Store education charts and Apple has recently declared it the application of the day.

Today, we employ more than 150 people and have offices in Zagreb and San Mateo, California. We are constantly expanding what Photomath can do and the types and quality of our content - we have so much more ahead of us and we need your help to get there.

The role of the AI team is to tackle seemingly unsolvable problems using cutting-edge technology, focused on the solutions based on the fast-changing and always stunning field of deep learning. We would like to enhance our end-to-end AI team where 20 dedicated engineers cover all development activities from data engineering to model training and model serving APIs.

Team’s tech stack:

  • Python (PyTorch, TensorFlow, numpy, FastAPI)
  • Docker, Kubernetes, GCP
  • PostgreSQL


What is the role?

As a Senior Machine Learning Engineer in our AI team, you’ll be:

  • Owning, researching, developing and evaluating AI models for all current and future complex and unfamiliar product features
  • Setting and maintaining course of AI projects from the technical perspective
  • Researching and selecting the right tools for the job, keeping up-to-date with new developments in the ML field
  • Communicating with software and data engineers on the subjects of datasets, pipelines and model testing and deployment
  • Troubleshooting and finding sustainable solutions to data issues and anomalies
  • Proactively identifying opportunities to improve the data and/or serving infrastructure
  • Providing guidance and mentorship to junior engineers

Our ideal candidate has:

  • 5+ years of relevant industry experience building machine learning models
  • Experience with Pytorch and/or TensorFlow
  • Experience with the numpy-scipy stack
  • Experience training deep learning models on computer vision tasks
  • Ability to read and interpret latest ML research papers and adapt their implementations
  • Solid understanding of statistics and probability theory
  • An up-to-date insight into state-of-the art ML models for major CV and NLP tasks
  • Knowledge to get and manipulate data from a relational database using SQL
  • Solid understanding of code complexity and performance
  • Familiarity with some of the ML model serving frameworks (TorchServe, TF Serving, Triton)
  • Understanding of the fundamental and unique challenges behind productionizing ML algorithms
  • Willingness to learn and implement new trends and technologies
  • Ability and willingness to work in a team, accept feedback, help younger colleagues and maintain an open and friendly atmosphere

An additional plus if you have:

  • Experience with a mass-market consumer-facing product
  • High quality and easy to maintain code running in production
  • Experience in developing SaaS products and RESTful API's
  • Experience in other relevant programming languages (Go, Kotlin, Java, C++)

Salary range: 

  • 26.000 - 31.000 kn gross, depending on candidate's experience with possible exceptions
  • Stock options 

What we offer:

  • An opportunity to work on innovative AI projects with global impact in a team of top AI engineers
  • An opportunity to work with the cutting edge machine learning models (i.e. Transformer-based architectures for tasks combining computer vision and natural language processing)
  • An opportunity to make an impact and define processes and ways of working in the fast-growing team
  • Experimenting with new technologies
  • Flexible working hours and work from home arrangements with the possibility of full remote work 
  • Dedication to a healthy work-life balance and various benefits for parents 
  • A diverse environment with agile and talented individuals across the career spectrum - to teach and be taught
  • A friendly, collaboration-heavy team atmosphere
  • A culture that recognizes and rewards dedication and success
  • Dedicated person (mentor/buddy) to help you navigate your first weeks in a new role
  • Dedicated Engineering Manager/Team lead with a strong emphasis on assisting your personal and professional growth (weekly 1:1s, clear expectations, mutual feedback, mentorship, clear and fair career progression track ect.)
  • Learning and growth opportunities through knowledge sharing, education and conferences, individual development plan with a dedicated budget, weekly time devoted to learning new things
  • Cutting edge hardware and equipment, budget for additional equipment
  • Company events and celebrations, company retreat, team budget for team building activities
  • Birthday and holiday presents for employees and their kids
  • Generous vacation and paid leave policy, sick leave without a doctor's note, annual physical exam (check-up)
  • Multisport card for various discounts at sport facilities
  • Underground bicycle parking garage
  • Modern office design, great view :) and great location (Zagreb, Strojarska 20)


Senior Machine Learning Engineer

Job description

Senior Machine Learning Engineer

Personal information
Professional data