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

Upto $200K/Year | Inpatient 7-On/7-Off | Dual-Certified PMHNP/FNP | Remote

Lead M&A Delivery

CAD Implementer (REMOTE-US)

Scientific Account Manager - Stem Cell Solutions- REMOTE

Regional Safety Leader - Remote Power & Energy

Sr. Program Manager (Healthcare)

Senior/Lead Software Engineer

Hubspot Marketing Consultant (Remote)- CONTRACT

Remote Sales Administrative Assistant

Remote eBay Expert

Electrical or Mechanical Commissioning Agent (Travel Required/Remote)

Senior Big Data Engineer: Snowflake + dbt (Remote)

Remote Senior Bioinformatics Manager, Genomics & Clinical

Business Process Analyst at General Dynamics: 100% Remote (LA)

Solution Architect – Agentic AI

NetSuite ARM Implementation Consultant - Senior Associate

PMT Engineer - Analytics and Automation

Payments & Product Partnerships Counsel, Americas

Financial Consultant I - Principal Connection

Dist Design Project Coord - Sr

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

Global Total Rewards Lead

Remote Payroll Specialist | Flexible Hours & Growth

Entry-Level Auto Technician: Hands-On Tire & Maintenance

Senior Associate, Risk and Structuring Analytics

Services Project Coordinator

Software Developer/Senior Software Developer

Manager, Professional Services- Streaming Media

Varonis Careers - Technical Account Manager

Manager, Accounts Payable

Monday.com Consultant[remote]

Remote Electrical Engineer — Critical Power Systems

Platform Engineering & Reliability Specialist (Azure)

Remote Patient Scheduling Specialist (Part-Time, 20hrs/wk)

Remote Retirement Inside Sales Specialist

Remote SAP MM Consultant – Entry Level

Client Support Specialist - Greater Philadelphia, PA Area (REMOTE)

Customer Service Specialist I at Aon Plc: 100% Remote (Fort Washington, PA, US)

Services Project Coordinator

Appeals Nurse Consultant (Remote)

Servicing Travel Advisor III - Chase Travel (Remote - Tempe, AZ)

Programmatic Solutions Manager, Uber Advertising

Manager, Flight Maintenance - Traveling

Customer Service / Pizza Maker (01068) 28803 Northwestern Hwy, Southfield, MI

Guidewire Architect - PC/CC/BC

Notary Stamp Order Online | West College Corner, IN

Senior Analyst, Program Cost Controls (Remote)

Sr. Tech Business Developer, Just Walk Out Technology by Amazon - AWS Solutions

Medical Consultant

SENIOR DIGITAL SALES EXECUTIVE (REMOTE/USA) - KSLA (GDM - GRAY MEDIA GROUP)