What are the responsibilities and job description for the Senior ML Research Engineer position at Drafted?
About Drafted
Drafted is unlocking creativity in the physical world. We’re building foundational models and generative pipelines that create floor plans and renderings instantly, so anyone can imagine their dream home. Starting with single-family homes, we plan to verticalize across all dimensions of the pre-construction stack.
Generative architecture is a prime domain for applied research with abundant data, verifiable constraints, and a clear value proposition. Our team of second-time founders, engineers, and designers pairs exceptional product taste with deep technical rigor to turn real-world buildability constraints into an intuitive, creative experience.
We're supported by investors like Patrick Collison (Stripe Co-Founder), Evan Moore (Co-Founder of Doordash), Josh Buckley, Jack Altman, and other, as well as advised by Sam Altman and David Holz.
Drafted's Values
We'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 Role
As an ML research engineer, you’ll contribute meaningfully towards advancing our model capabilities and bringing them into production. Our product blends exceptional product craft with deep technical work, so you’ll collaborate closely with both the engineering and research teams.
Example Projects
- 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
Desired Skillset
- Writing production serving code for LLMs / Diffusion models with PyTorch
- Written multi node training code for LLMs / diffusion models with PyTorch
- Bonus if you've written and tuned performance with custom Triton / CUDA kernels
- 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