Conduct groundbreaking research in neural network architecture design to advance state-of-the-art alternative architectures like state space models. Design novel architectures to improve model quality, inference efficiency, and adaptability across cloud and on-device environments. Requires deep expertise in architecture design, a proven research track record in top-tier ML venues, and proficiency in PyTorch or TensorFlow.
Key Highlights
Pioneering alternative architectures including State Space Models (SSMs) and efficient transformers
Focus on balancing model quality, latency, energy efficiency, and scalability
Translating architectural research into scalable real-world systems
Searching for Devops roles that provide visa sponsorship? Connect with international employers through Devops Jobs with Visa Sponsorship opportunities actively seeking talented professionals.
Key Responsibilities
Conduct groundbreaking research in neural network architecture design to advance the state-of-the-art in alternative architectures.
Design novel architectures that improve model quality, inference efficiency, and adaptability across diverse deployment environments.
Explore and develop capabilities such as statefulness, long-range memory, and innovative conditioning mechanisms.
Investigate how architectural decisions impact model trade-offs, including scalability, robustness, latency, and energy efficiency.
Develop new frameworks and tools to evaluate architectural innovations and benchmark performance.
Collaborate with cross-functional teams to translate architectural research into scalable and impactful systems.
Technical Skills Required
Neural Network Architecture Design
PyTorch
TensorFlow
Explore our comprehensive directory of visa sponsorship jobs from employers worldwide who are ready to sponsor talented international professionals.
Benefits & Perks
Competitive base salary
Equity package
Medical, dental, and vision insurance
9 weeks paternity & 12 weeks maternity leave
401(k)
Commuter Allowance
Flexible PTO
Daily meals and snacks
Visa sponsorship support
Nice to Have
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
Job Description
Our mission is to architect AI that learns from and interacts with the world like humans do.We're pioneering the model architectures that will make this possible. Our founding team met as PhDs at the Stanford AI Lab, where we invented State Space Models or SSMs, a new primitive for training efficient, large-scale foundation models. Our team combines deep expertise in model innovation and systems engineering paired with a design-minded product engineering team to build and ship cutting edge models and experiences.
Want the full job description?
Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn
This is a short excerpt. All rights to the full description belong to its original publisher.
