Senior ML Research Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Drafted · 2 weeks ago

Senior ML Research Engineer

positionSan Francisco, CA
timeFull-time
remoteOnsite
senioritySenior Level
money$150K/yr - $300K/yr
date5+ years exp

Drafted is a company focused on unlocking creativity in the physical world by building foundational models and generative pipelines for creating floor plans and renderings. The ML Research Engineer will advance model capabilities and collaborate with engineering and research teams to bring these models into production.

Artificial Intelligence (AI)Interior Design
check
H1B Sponsor Likelynote

Responsibilities

Contribute meaningfully towards advancing our model capabilities and bringing them into production
Teach our ML model how to handle multiple layers of context and constraints to consider in the generation process (i.e. Structural engineering, Civil engineering, Electrical, etc)
Figure out how to add direct to vector as an output of the model
Use RL to increase the accuracy of the model
Teach our ML model how to generate multi-story outputs
Lower the cost of inference by distilling our model
Speed up our data labeling processes with computer assisted labeling

Qualification

PyTorchLLMs / Diffusion modelsCUDAMulti-node trainingProduction serving codeNovel techniques iterationData labeling processes

Required

5+ years experience coding
Writing production serving code for LLMs / Diffusion models with PyTorch
Written multi node training code for LLMs / diffusion models with PyTorch
Iterated on novel techniques for meaningfully improving some KPI for LLM / diffusion models
Provided code for reproducing said novel techniques if open sourced, otherwise worked with practitioners for productionizing the techniques

Preferred

Bonus if you've written and tuned performance with custom Triton / CUDA kernels

Benefits

1-2% equity

Company

Drafted

twittertwittertwitter
company-logo
Drafted.ai is an AI-powered platform that creates personalized home floor plans and layouts based on user inputs
dateFounded in 2025
location
San Francisco, California, USA
people-size2-10 employees
web-link
https://www.drafted.ai

H1B Sponsorship

Drafted has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2020 (1)

Funding

Current Stage
Early Stage
Total Funding
$1.65M
Key Investors
Convective Capital
2025-12-23Pre Seed· $1.65M
Company data provided by crunchbase