The Browser Company is building a better way to use the internet with Dia, a browser that aims to empower and support users by helping them grow, create, and stay curious. They are looking for a founding ML Engineering TLM to build their ML strategy and incorporate ML to advance the product vision, turning user context into high-utility, personal experiences.
Requirements
- 6+ years of experience training, 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)
- You have production experience with Python and have experience fine-tuning open-source LLMs and going beyond simple LoRA fine-tuning
- Swift for on-device inference
- Python for models and tooling
- Support both encoder and decoder model families across client and server
- Scale the system using fine-tuned open-source LLMs
- Experience fine-tuning open-source LLMs and going beyond simple LoRA fine-tuning
Responsibilities
- Define and build our ML strategy, sequencing bets that improve Dia’s assistant and measurable user outcomes.
- Prototype, architect, and ship LLM‑powered features; establish techniques to train models that improve over time and personalize experiences, partnering with Design and Product Engineering to balance quality, speed, and scale for real‑world use.
- Audit and evolve the ML stack and infrastructure - spanning Swift for on-device inference and Python for models and tooling - to support both encoder and decoder model families across client and server.
- Partner on privacy and security by working with Security and Infra on data stewardship, deployment strategies, and responsible scaling.
- Establish ML Developer Experience by building tooling and workflows for high quality data curation, experimentation (fine-tuning, RL, prompting), evals, and continuous training to improve the models that power Dia.
- Support and build a talented team of machine learning engineers, helping them grow both technically and with a product mindset through fast iteration cycles.
- Browser context engine: learn a unified embedding space that fuses app integrations, enterprise connectors, and on-device signals to enable low-latency retrieval, routing, and personalization
Other
- 3+ years mentoring and leading senior engineers with a track record of tech-leading critical work and setting a sustainable execution pace.
- You're able to execute on critical projects, own large, complex codebases and drive initiatives within your team.
- You love to keep up with the latest advancements in AI and are excited to apply the latest models and techniques to push the boundaries of what's possible in a browser
- You’re pragmatic, motivated by nebulous problems, and excited to work in a startup environment with quick product validation cycles.
- We’re primarily focused on hiring in North American time zones and require that folks have 4+ hours of overlap time with team members in Eastern Time Zone.