Software Engineer, SDK - US (Remote)

This Software Engineer, reporting to the SDK Team Manager, will be responsible for building the software development kit (SDK) which helps our users accelerate their machine learning projects and build better models faster. The role involves exploring the pain points experienced by Machine Learning Engineers who are building state‑of‑the‑art models, designing solutions to address these issues, and implementing efficient and intuitive interfaces which integrate cleanly into the latest Machine Learning frameworks. The role involves working at the intersection of software development, machine learning, and novel product development.

Here are some key characteristics that will help you thrive in this role:

Outgoing and friendly: You'll love this role if you enjoy connecting with real users day to day, helping them solve issues and understand good patterns for using our tools. Day to day you'll be answering questions and requests with a kind, thoughtful tone that makes users feel appreciated and connected to our team.

Autonomous: If you work well in a self‑directed environment, and proactively find ways to improve processes and collaborate with team members or engaged users, your initiative will really shine in this role.

Curious and driven: Explore machine learning and learn more about the engineering stack and common ML workflows. Solve problems in both fast‑paced, short term sprints and in larger, more long‑term projects.

Organized: A core part of engineering support at Weights & Biases is organizing feedback from many channels into a single, orderly stream. Your organization skills and time management will be key to running this process well.

Responsibilities

  • Collaborate with senior engineers to design and build a multi‑platform / multi‑language Machine Learning SDK in a public github repository
  • Build out support for advanced machine learning workflows, such as distributed training
  • Build, extend, and enhance a highly reliable and performant data ingest SDK
  • Design thoughtful solutions to capture user issues in different client setups, and capture the pertinent info for debugging
  • Creatively improve on current features, and seek out ways to simplify the API and streamline the user experience
  • Improve CLI and SDK interfaces, implement new integrations with popular ML frameworks as they're released
  • Keep up with the latest trends in the ML world and leverage existing tools and frameworks whenever necessary

Requirements

  • 4+ years of professional software development in Python or Go
  • Experience communicating and collaborating on cross‑functional projects
  • Creative problem solving, and a willingness to dig into details
  • Experience contributing to architecture and systems design
  • Experience with ML frameworks such as PyTorch or TensorFlow

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

$146,000 - $203,000 a year

The US base pay for this position ranges from $146,000 USD per year in our lowest geographic market up to $203,000 USD per year in our highest geographic market. Weights and 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.

#J-18808-Ljbffr
Back to blog

Other Jobs To Apply

Virtual Pharma Account Manager: High-Impact Territory Sales

Principal Business Development Manager Pharma at Waters Corporation: 100% Remote (US)

Dist Design Project Coord - Sr

Market Commentary Writer and Social Editor at Wealthfront: 100% Remote (San Francisco, CA, New York City, NY)

Analyst, Early Career (Full Stack)

Sales Executive, Digital Assets

Senior Pre-Sales Systems Engineer, Enterprise

Remote Enterprise Identity & Security AE

Remote Retirement Inside Sales Specialist

Account Executive, B2B SaaS — Hybrid/Remote

[Remote] Director - Special Investigations (FIU)

Senior Manager, SAP AI Engineering

Associate Director, eCommerce Operations, eProcurement

Manager, Professional Services- Streaming Media

Commercial Sales Engineer - East - Remote

Emerging Fraud Researcher

HubSpot Administrator - Remote

Remote Snowflake Data Lake Support Pro

Investigator, Operational Excellence

Remote Principal Engineer, Identity & Engagement

Crypto Twitter Content Lead: Real-Time & Witty

TikTok Social Programming, Bleacher Report at Warner Bros. Discovery: 100% Remote (NY, GA)

Inbound Sales Call Center Representative

Senior Manager, Compensation & People Analytics (Remote)

Virtual Sales Rep: Fully Remote, No Cold Calls, Great Training

Services Project Coordinator

Dashers - Sign Up and Start Earning

Remote-Eligible Senior Business Analyst: SDLC & Stakeholder Leader

Notary Stamp Order Online | West College Corner, IN

Remote Freelance Japanese Teacher & 1:1 Coach

Audio & Digital Account Executive

Assistant, Contract Administration

Coordinator, People Data Operations (Remote)

Live Entertainment Marketing Coordinator

Global Technical Product Advisor - Off Road

Guest Experience Center Agent / Inside Sales

Remote SEO Content Lead

Remote Diagnostic Technician: Fast, CustomerFocused Support

Navy Medicine Analytics SME — Remote

Digital Evangelist

Workday Solution Architect

Experience Strategy Manager

Part-Time Oncology Data Specialist - Certified (remote)

Technical Support Consultant Analyst (Remote - PST Working Hours)

GL Claims Specialist - Remote

Senior Underwriter - Management Liability

Controller / Director of Finance

Remote AR Analyst I: Collections & Resolution

Senior Sales Executive

Manager, Flight Maintenance - Traveling

Common Interview Questions And Answers

1. HOW DO YOU PLAN YOUR DAY?

This is what this question poses: When do you focus and start working seriously? What are the hours you work optimally? Are you a night owl? A morning bird? Remote teams can be made up of people working on different shifts and around the world, so you won't necessarily be stuck in the 9-5 schedule if it's not for you...

2. HOW DO YOU USE THE DIFFERENT COMMUNICATION TOOLS IN DIFFERENT SITUATIONS?

When you're working on a remote team, there's no way to chat in the hallway between meetings or catch up on the latest project during an office carpool. Therefore, virtual communication will be absolutely essential to get your work done...

3. WHAT IS "WORKING REMOTE" REALLY FOR YOU?

Many people want to work remotely because of the flexibility it allows. You can work anywhere and at any time of the day...

4. WHAT DO YOU NEED IN YOUR PHYSICAL WORKSPACE TO SUCCEED IN YOUR WORK?

With this question, companies are looking to see what equipment they may need to provide you with and to verify how aware you are of what remote working could mean for you physically and logistically...

5. HOW DO YOU PROCESS INFORMATION?

Several years ago, I was working in a team to plan a big event. My supervisor made us all work as a team before the big day. One of our activities has been to find out how each of us processes information...

6. HOW DO YOU MANAGE THE CALENDAR AND THE PROGRAM? WHICH APPLICATIONS / SYSTEM DO YOU USE?

Or you may receive even more specific questions, such as: What's on your calendar? Do you plan blocks of time to do certain types of work? Do you have an open calendar that everyone can see?...

7. HOW DO YOU ORGANIZE FILES, LINKS, AND TABS ON YOUR COMPUTER?

Just like your schedule, how you track files and other information is very important. After all, everything is digital!...

8. HOW TO PRIORITIZE WORK?

The day I watched Marie Forleo's film separating the important from the urgent, my life changed. Not all remote jobs start fast, but most of them are...

9. HOW DO YOU PREPARE FOR A MEETING AND PREPARE A MEETING? WHAT DO YOU SEE HAPPENING DURING THE MEETING?

Just as communication is essential when working remotely, so is organization. Because you won't have those opportunities in the elevator or a casual conversation in the lunchroom, you should take advantage of the little time you have in a video or phone conference...

10. HOW DO YOU USE TECHNOLOGY ON A DAILY BASIS, IN YOUR WORK AND FOR YOUR PLEASURE?

This is a great question because it shows your comfort level with technology, which is very important for a remote worker because you will be working with technology over time...