Staff Software Engineer, AI

Summary

Join The Browser Company in building a better way to use the internet by creating a browser that empowers and supports users. As a Machine Learning Engineer, you'll work alongside product engineers, designers, and the Co-Founder to build an LLM-powered interface for the internet.

Requirements

• 5+ years of experience optimizing and productionizing modern machine learning models, especially ones that run in a real-world product environment (bonus if you’ve worked closely with transformer models)
• You have production experience with a modern coding language like Python
• You have experience with fine-tuning, distilling, and improving modern machine learning infrastructure and models
• You're passionate about performance and efficiency and coming up with creative approaches to building a new kind of browser
• You have experience independently running critical projects or initiatives with minimal guidance
• You’re pragmatic, motivated by nebulous problems, and excited to work in a startup environment with quick product validation cycles

Responsibilities

• Scope and spearhead projects to fine-tune, distill, or train transformer models for various features within the browser
• Innovate our on-device model architecture in MLX, ONNX, and other frameworks and model formats
• Trial and improve new on-device models, including fine-tuning new LLMs, to be performant on a variety of machines
• Build infrastructure to collect or generate training data for building or improving models in a privacy safe way
• Create ways for us to determine and track model performance and accuracy to improve our app efficiency overtime

Benefits

• Comprehensive benefits package with employee medical, dental, and vision - we cover 100% of premiums for employees, and up to 95% for dependents
• 401k plan
• Flexible vacation policy - on average, our team members take between 15-20 vacation days a year, plus federal holidays (holidays vary by location)
• Remote-friendly working environment - our core working hours are 11 AM-2 PM Eastern Time
• 12 weeks of paid parental leave
• $1,500 USD home office stipend
• Employees based in the US also receive additional services like free annual memberships to One Medical (where available), Talkspace, Teladoc, and HealthAdvocate