AI Engineer - San Francisco

San Francisco, California
Marketing – Growth - ML /
Full-time /
At Weights & Biases, our mission is to build the best tools for AI developers. We founded our company on the insight that while there were excellent tools for developers to build better code, there were no similarly great tools to help ML practitioners build better models. Starting with our first experiment tracking product, we have since expanded our solution into a comprehensive AI developer platform for organizations focused on building their own deep learning models and generative AI applications.

Weights & Biases is a Series C company with $250M in funding, over 200 employees, and a rapidly growing user base. Our platform is an essential part of an AI builder’s daily work, serving research labs like OpenAI, MetaAI, MistralAI, and, as well as large enterprise teams such as NVIDIA, Salesforce, Genentech, Cisco, Fidelity, Samsung, Walmart, and more.


    • Actively engage with the developer community in San Francisco by organizing and attending GenAI-focused events and hackathons.
    • Develop both internal and external GenAI-powered tools to demonstrate the capabilities of Weave.
    • Create compelling GenAI blog posts and code examples to highlight Weave’s potential to a global audience.
    • Contribute to the development and maintenance of our open-source Weave integrations.
    • Help build and maintain our open source Weave integrations
    • Engage with the developer community on social media and discord
    • Collaborate closely with our Product team, providing feedback and testing new Weave features.


    • Located in the San Francisco Bay Area
    • 3+ years of industry experience in software engineering or machine learning engineering in a production environment
    • Experience building GenAI-powered tools for industry use or as personal side projects.
    • Proven track record of writing blog posts and/ or social media threads that explore and push the limits of LLM systems, ideally with GitHub code examples.
    • A curious and open mindset, with a willingness to experiment, learn, and iterate quickly.
    • High agency and the ability to work autonomously in the dynamic environment of a fast-growing startup.
    • A proactive, kind, and collaborative mindset

Preferred Qualifications

    • 5+ years of industry experience
    • Experience pre-training / fine-tuning LLMs at the 1b+ scale or training / fine-tuning multi-modal models like stable diffusion etc.
    • Experience working in a developer tools company
    • Have attended 3+ San Francisco hackathons in the past 12 months

Our Benefits

    • 🏝️ Flexible time off
    • 🩺 Medical, Dental, and Vision for employees and Family Coverage
    • 🏠 Remote first culture with in-office flexibility in San Francisco
    • 💵 Home office budget with a new high-powered laptop
    • 🥇 Truly competitive salary and equity
    • 🚼 12 weeks of Parental leave (U.S. specific)
    • 📈 401(k) (U.S. specific)
    • Supplemental benefits may be available depending on your location
    • Explore benefits by country
$182,000 - $248,000 a year
The US base pay for this position ranges from $182,000 USD per year in our lowest geographic market up to $248,000 USD per year in our highest geographic market. Weights & Biases is committed to providing competitive salary, equity, and benefits packages for all full-time employees. Individual compensation will be commensurate with the candidate's experience, qualifications, and geographic location.
We encourage you to apply even if your experience doesn't perfectly align with the job description as we seek out diverse and creative perspectives. Team members who love to learn and collaborate in an inclusive environment will flourish with us. We are an equal opportunity employer and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. If you need additional accommodations to feel comfortable during your interview process, reach out at