AI Engineer

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 and over 200 employees. We proudly serve over 1,000 customers and more than 30 foundation model builders including customers such as OpenAI, NVIDIA, Microsoft, and Toyota.

As an AI Engineer on the AI Growth team, you will play a crucial role in developing open-source integrations, building GenAI-powered tools, and engaging with the AI developer community to drive adoption of Weave. This role is ideal for someone who is deeply involved in the GenAI space, passionate about developer tools, has an interest in connecting with the community, and excited to contribute to a fast-growing product.

This is a highly technical and strategic role where you will have the opportunity to work closely with engineering, product, and research teams across our global team while also playing a key role in growing Weave’s adoption within India’s fast-moving GenAI ecosystem. If you are passionate about LLMs, fine-tuning, developer tools, and fostering AI communities, we’d love for you to join us in shaping the future of Weave.

Responsibilities:

• This role will see a balance of 70% working in code and 30% working with the community and events in the India market

• Contribute to the development and maintenance of our open-source Weave integrations.

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

• Collaborate closely with our Product team, providing feedback and testing new Weave features.

• Actively engage with the developer community in India by organizing and attending GenAI-focused events and hackathons.

• Help build out an ambassadors program in India to help us grow Weave with community support.

• Engage with the developers where they live online (X, LinkedIn, Discord, etc.)

Requirements:

• Based in India. This role will build the awareness of W&B in the booming India market while partnering with internal teams globally.

• 4-7+ years of industry experience

• Track record of open-source contributions.

• Track record of building GenAI applications.

• 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

• Experience planning and orchestrating events within the AI community

Strong Plus:

• Track record of organizing AI events or speaking at AI events.

• Planned and orchestrated 3+ events within the AI community in the last 12 months

• Strong know-how of Python and familiarity with ML stack and frameworks.

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

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 careers@wandb.com.

#LI-Remote