About the role
structured by ORICareersSenior ML EngineerOverviewOverviewapplicationDrafted's ValuesWe're a small team working fully in-person in San Francisco. We value high-ownership builders who want to be a part of a talented, highly motivated team.
What you will do
- Contribute meaningfully to building core infrastructure for training and running our models in production.
- Collaborate closely with both the engineering and research teams.
- 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.
What they are looking for
- Writing production serving code for diffusion models with PyTorch.
- Written multi node training code for diffusion models with PyTorch.
Nice to have
- Bonus if you've written and tuned performance with custom Triton / CUDA kernels.
Full posting text
CareersSenior ML EngineerOverviewOverviewapplicationDrafted's ValuesWe're a small team working fully in-person in San Francisco. We value high-ownership builders who want to be a part of a talented, highly motivated team. We're guided by the following values:Own the mission. We take agency, act like owners, and see problems through to real outcomes.Build in the open. We value direct feedback, fast learning, and growth through honest collaboration.Move with care and speed. We iterate quickly while staying deeply respectful of our teammates.Seek the why. We challenge assumptions, think from first principles, and never stop asking questions.Design for everyone. We believe anyone should be able to design and build a home they love.Solve what matters. We embrace hard problems and create new paths forward when none exist.The RoleAs an ML engineer, you'll contribute meaningfully to building core infrastructure for training and running our models in production. Our product blends exceptional product craft with deep technical work, so you'll collaborate closely with both the engineering and research teams.Example ProjectsTeach 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.Desired SkillsetWriting production serving code for diffusion models with PyTorch.Written multi node training code for diffusion models with PyTorch.Bonus if you've written and tuned performance with custom Triton / CUDA kernels.Apply NowRole DetailsLocationSan FranciscoEmployment TypeFull timeLocation TypeIn personSalary$150k-$300k + 0.5%-1.5%