Weights & BiasesWB

AI Engineer - India

At Weights & Biases our mission is to build the best tools for machine learning. Our experienced technical cofounders built Figure Eight, and our tools are being used by cutting-edge machine learning teams including OpenAI and Toyota. We're looking for passionate, driven people who love to learn and collaborate in an inclusive work environment.

Weights & Biases

Employee count: 201-500

India only
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 WB 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 [email protected].

About the job

Apply before

Posted on

Job type

Full Time

Experience level

Mid-level

Location requirements

Hiring timezones

India +/- 0 hours

About Weights & Biases

Learn more about Weights & Biases and their company culture.

View company profile

We're building a culture where people love coming to work to build cool things and collaborate with other smart people. It’s important to us that everyone feels comfortable being themselves and being honest with each other, and all of us take responsibility for making our customers successful. We love to work with people who are driven, kind and proactive. We care about diversity and if you are part of an underrepresented minority, we especially encourage you to apply to work here.
We are contributing to responsible AI with tools for transparency and reproducibility in model development and deployment.

Benefits

As second time founders, our company is small but well funded by top tier investors. We offer market salary, benefits and equity comparable to a Series C startup. When we're not all working from home, our sunny, pet-friendly office is located in the SOMA district of San Francisco.

Employee benefits

Learn about the employee benefits and perks provided at Weights & Biases.

View benefits

Paid parental leave

🚼 12 weeks of parental leave.

Home office budget

$500 to set yourself up for work from any home office.

Retirement benefits

🏦 401(k) to set you up for a future in saving for yourself.

Company meals

🍏 Come visit us in SF and nosh on snacks and team meals together.

View Weights & Biases's employee benefits
Claim this profileWeights & Biases logoWB

Weights & Biases

View company profile

Similar remote jobs

Here are other jobs you might want to apply for.

View all remote jobs

22 remote jobs at Weights & Biases

Explore the variety of open remote roles at Weights & Biases, offering flexible work options across multiple disciplines and skill levels.

View all jobs at Weights & Biases

Remote companies like Weights & Biases

Find your next opportunity by exploring profiles of companies that are similar to Weights & Biases. Compare culture, benefits, and job openings on Himalayas.

View all companies

Find your dream job

Sign up now and join over 85,000 remote workers who receive personalized job alerts, curated job matches, and more for free!

Sign up
Himalayas profile for an example user named Frankie Sullivan