ML Tech Lead (Search team)
We are currently looking for the ML/AI Tech Lead to join the Search team. You will be responsible for improving core search algorithms, including advancing the system’s understanding of user queries and increasing the relevance of results. The role also involves designing and developing new features aimed at enhancing the overall search experience.
Key focus areas
Driving the application of modern AI/ML models to improve search relevance (a major strategic direction)
Integrating AI-driven features into user flows, from prototyping to production-grade solutions
Embedding AI capabilities into product experiences to improve both efficiency and user satisfaction
Tech stack: Go, Python, Scala, Spark, modern LLMs, PyTorch, Docker, Kubernetes, Airflow, MongoDB, Cassandra, Kafka, Jenkins
Our engineers are responsible not only for data analysis, hypothesis creation, modeling, and experimentation, but also for writing production code capable of handling high loads.
Responsibilities
Strong algorithmic skills: knowledge of fundamental algorithms and data structures, as well as their implementation in standard libraries
Proficiency in any language: experience in developing and debugging distributed, high-load applications, and writing testable code
Industry experience in Machine Learning: familiarity with key modeling and training approaches (linear models, GBDT, DNN, etc.), knowledge of statistics, and end-to-end ML development experience, including raw data analysis, hypothesis formulation, model creation, experimentation, and deploying results into production
Participate in new feature and product development discussions
Managing and mentoring an ML team of 3-5 engineers
Preferred
Experience in developing recommendation or search systems
Experience training neural network models
Experience working with various data sources: Spark, Cassandra, Mongo, Kafka
Knowledge of Go
We offer
Compensation package: base salary and performance-based bonuses
Office-first: flexible hours with a possibility to work remotely 52 days per year
Care & Wellbeing: health insurance (including dental care) for employees and their children, daily meal allowance, and 100% paid sick leave
Relocation Support: full assistance for a smooth relocation for the employees and their family
Team & Growth: collaboration with colleagues across Portugal, Brazil, Latvia and China, with opportunities for promotions, professional trainings, and English & Portuguese courses
Community & Engagement: annual team building activities, knowledge-sharing workshops, and a strong sense of team work
Published on: 5/12/2026

Joom Group
Joom Group is an international tech-centric group of e-commerce companies founded in 2016 in Latvia. We are here to transform the largest industry in the world, global trade, making it more transparent, efficient, and technology-driven.
Please let Joom Group know you found this job on Wantapply.com. It helps us to get more jobs on our site. Thanks!




