Technical Lead Manager, Machine Learning

Hi, we’re The Browser Company 👋 and 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.

About The Role

As our founding ML Engineering TLM at the Browser Company, you’ll build our strategy and define how we incorporate ML to advance Dia ’s product vision. The browser holds most of the context about your workday, from memory of the tasks you’re trying to accomplish to access to all of your web apps. We use ML and AI to turn that context into high‑utility experiences that feel personal and improve over time, grounded in capabilities like computer‑use, memory, and web‑app integrations.

Dia’s Assistant doesn’t just execute actions, it also suggests and recommend tasks, and learns which choices drive real outcomes. We leverage modern training techniques such as SFT, RL, and GEPA to optimize these experiences while protecting privacy. To move fast and learn continuously, we’re building a first‑class ML developer environment: curated data pipelines, tight experiment loops, offline/online evals with telemetry, and one‑click deploys with strong observability.

Our Management Philosophy

Engineering Managers at Browser Company are hands‑on technical leaders who build high‑performing and psychologically‑safe teams with a diverse group of individuals. You’ll work closely with your team to make product decisions, prioritize work, ship features, and promote an engineering culture of knowledge sharing and mentorship. Engineering Managers follow the same onboarding pathway as an IC engineer to learn our product data, and build processes, then ramp into leadership.

Overall you will...

• Define and build our ML strategy , sequencing bets that improve Dia’s assistant and measurable user outcomes. You’ll stay ahead of the latest AI advancements and translate them into strategic opportunities for Dia’s roadmap.

• 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. Over time, scale the system using fine‑tuned open‑source LLMs.

• 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. You’ll play a key role in contributing across the engineering org as we scale—owning processes, recruiting, and onboarding—and proactively improve architecture and practices to enhance performance, stability, and maintainability.

Technical Projects You’ll Shape With Us…

• 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

• Personalize the command bar: develop ranking, intent understanding, and context‑aware suggestions; fine‑tune and evaluate models to measurably boost relevance and engagement

• Evolve our data flywheel by defining how we instrument product signals, design collection/storage pipelines, create labeling/evaluation loops, and continuously retrain to improve feature quality and personalization.

Qualifications

• 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)

• 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 have production experience with Python and have experience fine‑tuning open‑source LLMs and going beyond simple LoRA fine‑tuning

• 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.

Compensation and Benefits

• Our total compensation package for full‑time employees includes base salary, equity, and benefits. The annual salary range for this role is $300,000- $360,000 USD . The actual salary offered will vary based on experience level and interview performance.

• Benefits: We also offer a wide range of perks and benefits designed to support you, your family and to help you engage with your local community. To learn more, visit go.atlassian.com/perksandbenefits

• Location: We’re a remote‑friendly company and can hire in any country where Atlassian has a legal entity. If you live in New York (or want to visit), you’re welcome to work from our beautiful office in Williamsburg.

The Browser Company is an ambitious team of close to 100 people (and growing!) who are passionate about building great products. We are a remote‑first, distributed team, with the option to work from office in Brooklyn, New York. We strongly support diversity and encourage people from all backgrounds to apply.

🚙 To read more about what we value as a company, check out Notes on Roadtrips on our blog.