Technical Lead, Machine Learning Engineer

Summary

Join The Browser Company and become a Technical Lead, Machine Learning Engineer, working on Dia, the next LLM-powered interface for the internet. Collaborate with a diverse team to fine-tune LLMs, optimize on-device and cloud-based inference, and build AI-powered features. You will build evaluation pipelines, improve datasets, and prototype new features. The role requires 8+ years of experience in optimizing and productionizing ML models, expertise in fine-tuning LLMs, and experience with a modern coding language like Python. The company offers a flexible compensation model with a salary range of $250,000 - $300,000, comprehensive benefits, flexible vacation, remote-friendly work environment, and paid parental leave.

Requirements

• 8+ years of experience optimizing and productionizing modern ML models, especially ones that run in a real-world product environment (bonus if you’ve worked closely with transformer models)
• Deep experience fine-tuning open-source LLMs and going beyond simple LoRA fine-tuning
• Production experience with a modern coding language like Python
• Passion for on-device performance and excitement to push the boundaries of what's possible in a browser
• Experience independently running critical projects, shipping ML features, and leading initiatives with minimal guidance
• Pragmatism, motivation by nebulous problems, and excitement to work in a startup environment with quick product validation cycles
• 4+ hours of overlap time with team members in Eastern Time Zone

Responsibilities

• Fine-tune, distill, and optimize LLMs to improve performance, reduce latency, and enhance efficiency for on-device and cloud-based inference
• Improve our on-device model architecture, leveraging frameworks like MLX, ONNX, and TFLite to ensure models run efficiently across different devices
• Experiment with and integrate new LLMs, fine-tuning them for specific browser-based use cases while balancing quality, speed, and resource constraints
• Build evaluation pipelines to track model performance, accuracy, and real-world effectiveness over time
• Collaborate with product ops teams to build and improve datasets that accurately match product needs
• Collaborate with product engineers and designers to prototype and ship AI-powered features that enhance user experience
• Optimize inference strategies, including running models on-device, in the cloud, or in hybrid configurations to maximize throughput and resource usage

Preferred Qualifications

Experience working closely with transformer models

Benefits

• Flexible compensation model with the ability to choose the cash-to-equity ratio
• Annual salary range of $250,000 - $300,000 USD
• Comprehensive benefits package with employee medical, dental, and vision - 100% of premiums covered for employees, and up to 95% for dependents
• 401k plan
• Flexible vacation policy - on average, 15-20 vacation days a year, plus federal holidays
• Remote-friendly working environment - core working hours are 11 AM-2 PM Eastern Time
• 12 weeks of paid parental leave
• $1,500 USD home office stipend
• Free annual memberships to One Medical (where available), Talkspace, Teladoc, and HealthAdvocate (for US-based employees)