Machine Learning Engineer - Senior Software Engineer

UG Labs is looking for a Senior Machine Learning Engineer to join our R&D team. In this role, you will help us implement a conversational engine for kids. An ideal candidate will have experience in leading challenging technical projects, ML engineering, software architecture and backend development in Python.

Key Responsibilities

• Integrate and deploy ML algorithms to production under strong scale and SLA considerations. 
• Design and implement our gaming and toy platform.
• Collaborate with research, engineering and product stakeholders.
• Own critical parts of our software infrastructure, delivering end-to-end solutions.

What are we looking for?

• Excellent problem-solving skills with the ability to deliver quick and high-quality solutions.
• Excellent Python development skills.
• Experienced in software development, on the level you’d expect from a SW Tech Lead / Team Lead.
• Proficiency in MLOps / ML Engineering.
• Proficiency in software performance optimization.
• Good knowledge of contemporary ML models (LLMs, Diffusion Models).

What would help you stand out?

• Track record of serving and scaling self-hosted ML workloads in production.
• Experience with optimizing PyTorch inference code performance.
• Excellent interpersonal skills and ability to transfer knowledge to other team members.

Our Tech Stack

Cloud: GCP, Lambda Labs, Azure

Backend: Python (gRPC/HTTP), PostgreSQL, Docker

DevOps / Infra: Terraform, Github Actions

Frontend: Unity, Flutter

About UG Labs

UG Labs has developed a novel way to infuse voice interactivity and conversational AI into games and toys. The new technology comes with a groundbreaking design paradigm and mechanics that focus on younger users but allow deeper immersion and engaging experiences for the entire family.

UG’s game engine incorporates the latest speech recognition and language models based on proprietary algorithms and data developed and trained by UG. The engine allows authentic open-ended conversations that affect the game flow and experience, allowing unscripted and on-the-fly exploration that works.