Tech Lead, Ello

• You should have a strong background in software engineering and machine learning. We work on research problems on a production timeline. You will be successful in this role if you enjoy setting up scalable training systems and infrastructure from scratch and like shipping real-world products
• Most importantly, you are excited about the opportunity to make a real impact on child development at scale
• 4+ years experience as an ML engineer/researcher working with modern Machine Learning toolkits (Python stack, very comfortable with PyTorch)
• 2+ years of experience building, scaling, and optimizing ML systems
• Experience setting up ML ops workflows and infrastructure from scratch in a startup environment (ideally in Terraform on AWS)
• Production programming experience on a data-intensive project in distributed infrastructure
• Experience deploying distributed data and training pipelines on Kubernetes
• Experience with large language models
• Experience writing high-performance, parallelized code
• Applied mathematics and linear algebra proficiency at the level at which you can contribute to research discussions with a deeply technical team
• Very accustomed to agile software engineering processes, git flow, etc

Desirable

• As the Tech Lead on the ML team, you will work closely with the Co-Founder & CTO to lead the architecture design for all of our ML efforts. You will be in charge of MLops in collaboration with our Platform team and set good standards for your colleagues on critical design questions around dataset management, experiment tracking, and distributed training infrastructure.
• You will also get into the weeds as an IC, develop and review model and application code, and implement state-of-the-art machine learning systems for reading instruction, story generation, and interactive dialogue teaching
• Oversee the development and deployment of machine learning models to production environments
• Lead by example, fostering a culture of excellence in architecture, coding, and model development
• Provide guidance, regular feedback, education, and mentoring to other members of the ML team
• Monitor and maintain deployed models and data pipelines
• Independently lead research work streams as part of Ello’s ML team
• Lead collaboration with platform and Unity engineers as well as TPM and data team on data collection, pipeline design, model training, and deployment
• Continuously learn and stay up to date with the latest advancements in machine learning engineering