As an AI Engineer, you will develop open-source integrations, build GenAI tools, and engage with the AI developer community while collaborating with global teams.
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 [email protected].
#LI-Remote
Top Skills
Genai
Llms
Ml Stack
Python
Stable Diffusion
Similar Jobs
Machine Learning • Software
The role involves providing technical support to customers, addressing their inquiries, and troubleshooting issues related to AI and machine learning products. Collaborating with engineering teams and maintaining communication SLAs are key responsibilities.
Top Skills:
AWSAzureColabsDockerGCPJupyter NotebooksKerasKubernetesMatplotlibPythonPyTorchPytorch LightningTensorboardTensorFlow
Artificial Intelligence • Blockchain • Fintech • Financial Services • Cryptocurrency • NFT • Web3
Design and develop test automation frameworks, implement automated test scripts, establish performance testing strategies, and provide technical leadership.
Top Skills:
AWSCi/CdJavaScriptPlaywrightPythonSeleniumUipath
Agency • Digital Media • eCommerce • Professional Services • Software • Analytics • Consulting
As a Mid-Level Backend Developer, you will design and maintain WordPress applications, manage DNS configurations, mentor juniors, and collaborate on scalable solutions.
Top Skills:
AWSCSSDockerGitGCPHTMLJavaScriptKubernetesMySQLPHPRestful ApisWordpressWpengine
What you need to know about the Pune Tech Scene
Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.
