Featherless AI · 1 day ago
Machine Learning Engineer — AI Architecture Research
United States
Full-time
Remote
Mid, Senior LevelFeatherlessAI is seeking a Machine Learning Engineer focused on AI architecture research to design, prototype, and validate next-generation model architectures. The role involves researching and developing new neural network architectures and collaborating with engineers to ensure efficient deployment of these architectures.
Artificial Intelligence (AI)Cloud ComputingDatabase
Responsibilities
Research and develop new neural network architectures (e.g. alternatives or extensions to Transformers, recurrent / hybrid models, long-context systems)
Design and run architecture-level experiments (scaling laws, memory mechanisms, compute trade-offs)
Prototype models end-to-end — from research code to training-ready implementations
Collaborate with inference and systems engineers to ensure architectures are deployable and efficient
Analyze model behavior, failure modes, and inductive biases
Read, reproduce, and extend cutting-edge research papers
Contribute to internal research notes, benchmarks, and open-source efforts (where applicable)
Qualification
Required
Strong background in machine learning fundamentals and deep learning
Hands-on experience implementing model architectures from scratch
Solid understanding of: Attention mechanisms, RNNs, state-space models, or hybrid architectures
Solid understanding of: Training dynamics, scaling behavior, and optimization
Solid understanding of: Memory, latency, and compute constraints at the model level
Comfortable working in PyTorch or JAX
Ability to move fluidly between theory, experimentation, and engineering
Clear communicator who can explain architectural trade-offs
Preferred
Experience with non-Transformer architectures (RNN variants, SSMs, long-context models)
Background in research-driven startups or open-source ML projects
Experience with large-scale training or custom training loops
Publications, preprints, or notable research contributions
Familiarity with inference optimization and deployment constraints
Benefits
Competitive compensation
Meaningful equity
Company
Featherless AI
We enable serverless inference via our GPU orchestration and model load-balancing system.
Founded in 2023
2-10 employees
H1B Sponsorship
Featherless AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)
Funding
Current Stage
Early StageTotal Funding
$5MKey Investors
Airbus Ventures
2025-10-31Series A
2025-03-17Seed· $5M
Company data provided by crunchbase