Job Description: Machine Learning Engineer

Employment Type: Full-Time Location Type: Remote Salary: Not Specified

About The Browser Company 👋

We're building a better way to use the internet. Browsers are unique in that they are one of the only pieces of software that you share with your parents as well as your kids. Which makes sense, they're our doorway to the most important things — through them we socialize with loved ones, work on our passion projects, and explore our curiosities. But on their own, they don’t actually do a whole lot, they’re kind of just there. They don’t help us organize our messy lives or make it easier to compose our ideas. We believe that the browser could do so much more — it can empower and support the amazing things we do on the internet. That’s why we’re building one: a browser that can help us grow, create, and stay curious.

To accomplish this lofty task, we’re building a diverse team of people from different backgrounds and experiences. This isn’t optional, it’s crucial to our mission, as we need a wide range of perspectives to challenge our assumptions and shape our browser through a bold, creative lens. With that in mind, we especially encourage women, people of color, and others from historically marginalized groups to apply.

About The Role

Browsers know everything about us and what we do everyday, yet they can’t predict our next move, morph themselves to better suit our tasks, or proactively reduce repetitive tasks during your work day. At The Browser Company, we’re changing that by building Dia.

As a Machine Learning Engineer, you’ll be working alongside ML engineers, product engineers, designers, and our cofounder and CTO, Hursh Agrawal, to build the next LLM-powered interface for the internet. You’ll work with designers and product engineers to define the product, fine-tune LLMs and smaller transformers like BERT, and iterate on our how we host models both in the cloud and on-device to improve latency and resource usage.

Overall You Will:

• 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.
• Set product direction via your expertise in ML and how it can help power Dia and increase its capabilities.
• Keep up with the latest advancements in AI and apply new models and techniques to improve Dia where applicable.
• Build evaluation pipelines to track model performance, accuracy, and real-world effectiveness over time. You'll also collaborate with product ops teams to build and improve datasets that accurately match product needs.
• Iterate on the product experience by integrating and improving the models used in Dia, working alongside product engineers and designers.

After 1 Month You Will:

• Onboard to the team and codebase with your onboarding buddy.
• Attend onboarding presentations about the company, product, codebase, and culture.
• Become familiar with the Swift language, the Dia codebase, and how we ship features.
• Ship a few bug fixes and small improvements across our product codebase and tooling.
• Have trained your first model, either improving an existing flow or enabling an entirely new one and have integrated it yourself into our product codebase.
• Have pair programmed with a few people on the engineering team.
• Be regularly posting product feedback about the browser in our #dogfooding channel.

After 3 Months You Will:

• Be familiar with how we prototype and build new features, working with product engineers to brainstorm ways to use models to add intelligence to Dia.
• Be familiar with our cloud infrastructure and data pipelines.
• Be familiar with how we run inference both on-device and in the cloud.
• Be testing new prototypes with existing, on-device models to test performance and viability.
• Participate in product brainstorms to think about the future of Dia.
• Be trained to interview candidates for roles at The Browser Company.
• Be contributing to... [Content truncated]