Your ImpactConduct groundbreaking research in neural network architecture design to advance the state-of-the-art in alternative architectures (e.g., state space models, efficient transformers, hybrid architectures).
- Design novel architectures that improve model quality, inference efficiency, and adaptability across diverse deployment environments, from cloud to on-device.
- Explore and develop capabilities such as statefulness, long-range memory, and innovative conditioning mechanisms for enhancing model expressiveness and generalization.
- Investigate how architectural decisions impact model trade-offs, including scalability, robustness, latency, and energy efficiency.
- Develop new frameworks and tools to evaluate architectural innovations, benchmarking performance across research and production settings.
- Collaborate with cross-functional teams to translate architectural research into scalable and impactful systems for real-world applications.
What You Bring- Deep expertise in architecture design, with experience in researching or deploying advanced architectures (e.g., state space models, transformers, RNN variants, CNN variants).
- Strong understanding of how architectures interact with system constraints, including deployment in cloud environments or on-device.
- Proficiency in designing architectures that balance quality, efficiency, and adaptability across different use cases and modalities (e.g., vision, audio, text).
- Familiarity with generative modeling paradigms like autoregressive and diffusion models, and designing capabilities such as statefulness and conditioning in deep learning models.
- A proven research track record in top-tier ML/AI venues (e.g., NeurIPS, ICML, ICLR, CVPR) or demonstrable contributions to state-of-the-art architectures.
- Exceptional analytical and problem-solving skills, with a focus on experimentation and iterative refinement.
- Strong programming skills in deep learning frameworks such as PyTorch or TensorFlow, and experience with profiling tools for understanding model performance.
Nice-To-Haves- Prior research or publications in state space models, efficient Transformers or other alternative architectures.
- Research or practical experience in designing architectures for multi-modal systems.
- Early-stage startup experience or a track record of rapid innovation in R&D environments.
More DetailsIn-office policy: We're an in-person team based out of offices in US San Francisco, GB London and I🇳 Bangalore. We love being in the office, hanging out together, and learning from each other every day.
Visa sponsorship: We provide visa sponsorship support and assess each circumstance on a case-by-case basis. However, visa sponsorship is dependent on many factors, including the role you are applying for, and the location you are going to be based, and so we can't always guarantee success. Your Recruiter will work with you to understand your visa sponsorship needs from the first call.
We ship fast. All of our work is novel and cutting edge, and execution speed is paramount. We have a high bar, and we don't sacrifice quality or design along the way.
We support each other. We have an open & inclusive culture that's focused on giving everyone the resources they need to succeed.
Our Benefits (US Employees Only) Compensation Competitive base salary alongside attractive equity package.
Health Insurance Fully covered medical insurance along with dental and vision for you and your family.
🧒🧒
Parental Leave 9 weeks paternity & 12 weeks maternity leave
401(k)🚆
Commuter Allowance A monthly stipend to help you get to and from the office.
Flexible PTO Take as much time as you need to recharge your batteries.
🍲
Meals & Snacks Lunch, dinner and plenty of snacks, provided daily.
🦖
Your own personal Yoshi 
