Location: New York,NY, USA
We're a team of machine learning engineers training task-specific generative models for psychology. Our goal is to build an AI therapist to help people change their mind and their lives in the ways that they want to. We partner with organizations around the globe and power use cases, including AI-assisted crisis text response, while securing best-in-class datasets to power our models. Success to us means every human being in need of support having somewhere to go. We're a well-funded, seed-stage startup backed by top-tier tech investors involved in Huggingface, ElevenLabs, Replit, Captions, Shopify, Plaid, Notion, Canva, Twitch, Airtable, and others We're building a powerful team by empowering our engineers with the autonomy, flexibility, and resources to do their best work. We dream big and iterate fast. If that sounds like home, we'd love to hear from you. The Role As ML Lead, you'll collaborate with the founders on ML research, data collection and curation, and be responsible for training large models from start to finish, in collaboration with our founding ML engineers. You'll be able to work at a faster pace than almost anywhere else, while writing high-quality code and producing meaningful scientific insights. You'll be primarily working with 70B parameter language models. You'll be involved in data collection, data curation, supervised fine-tuning, and preference optimization. Our research state-of-the-art research in reinforcement learning for language models. You will be responsible for reading papers and identifying state-of-the-art techniques for us to learn from. Our application backend is written in Kotlin and our ML stack (PyTorch) utilizes modern tooling in the ML space, including some that we've developed in-house. We write high-quality, typed, Zen code. About you: