JOB DETAILS
AI Researcher — AI Architecture Research
CompanyFeatherless.ai
LocationRemote
Work ModeRemote
PostedSeptember 20, 2026
About The Company
No description available for this Company.
About the Role
About the Role
We’re looking for an AI Researcher focused on AI architecture research to help design, analyze, and advance next-generation model architectures. You’ll work at the intersection of theory and production—publishing novel research while collaborating closely with engineers to turn ideas into real systems.
This role is ideal for someone who has published research papers and wants to see their work directly shape deployed models, not just benchmarks.
What You’ll Work On
Research and design novel AI architectures (e.g. alternatives to standard Transformer designs, long-context models, efficient sequence modeling, hybrid architectures)
Explore architectural improvements for scalability, efficiency, and stability
Prototype and evaluate new architectures through ablations, benchmarks, and empirical studies
Author and co-author research papers for top ML conferences and journals
Collaborate with engineering teams to translate research into training and inference systems
Stay current with state-of-the-art research and identify promising directions early
What We’re Looking For
Strong background in machine learning research, with a focus on model architecture
Publication record in ML/AI venues (e.g. NeurIPS, ICML, ICLR, COLM, ACL, EMNLP, arXiv)
Deep understanding of:
Neural network architectures
Sequence models and attention mechanisms
Training dynamics and optimization
Hands-on experience with PyTorch or JAX
Ability to reason rigorously, design clean experiments, and communicate results clearly
Comfortable working in a fast-moving startup environment
Nice to Have
Experience with non-Transformer architectures (e.g. RNN-based, state-space, hybrid models)
Work on long-context or memory-efficient models
Open-source research contributions
Experience bridging research and production systems
Background in efficient training or inference-aware architecture design
Why Join Us
High ownership over research direction and roadmap
Clear path to publishing impactful work
Tight feedback loop between research and real-world deployment
Small, highly technical team with strong research culture
Competitive compensation and meaningful equity
We’re looking for an AI Researcher focused on AI architecture research to help design, analyze, and advance next-generation model architectures. You’ll work at the intersection of theory and production—publishing novel research while collaborating closely with engineers to turn ideas into real systems.
This role is ideal for someone who has published research papers and wants to see their work directly shape deployed models, not just benchmarks.
What You’ll Work On
Research and design novel AI architectures (e.g. alternatives to standard Transformer designs, long-context models, efficient sequence modeling, hybrid architectures)
Explore architectural improvements for scalability, efficiency, and stability
Prototype and evaluate new architectures through ablations, benchmarks, and empirical studies
Author and co-author research papers for top ML conferences and journals
Collaborate with engineering teams to translate research into training and inference systems
Stay current with state-of-the-art research and identify promising directions early
What We’re Looking For
Strong background in machine learning research, with a focus on model architecture
Publication record in ML/AI venues (e.g. NeurIPS, ICML, ICLR, COLM, ACL, EMNLP, arXiv)
Deep understanding of:
Neural network architectures
Sequence models and attention mechanisms
Training dynamics and optimization
Hands-on experience with PyTorch or JAX
Ability to reason rigorously, design clean experiments, and communicate results clearly
Comfortable working in a fast-moving startup environment
Nice to Have
Experience with non-Transformer architectures (e.g. RNN-based, state-space, hybrid models)
Work on long-context or memory-efficient models
Open-source research contributions
Experience bridging research and production systems
Background in efficient training or inference-aware architecture design
Why Join Us
High ownership over research direction and roadmap
Clear path to publishing impactful work
Tight feedback loop between research and real-world deployment
Small, highly technical team with strong research culture
Competitive compensation and meaningful equity
Key Skills
PyTorchJAXNeural Network ArchitecturesAttention MechanismsMachine Learning Research
Categories
EngineeringData
Benefits
Equity
Job Information
📋Core Responsibilities
Research, design, and analyze novel AI architectures such as alternatives to standard Transformers, long-context models, efficient sequence modeling, and hybrid architectures; prototype and evaluate them via ablations and benchmarks, publish research papers, and collaborate with engineering to translate research into production training and inference systems.
📋Job Type
full time
📊Experience Level
Not specified
💼Company Size
Not specified
📊Visa Sponsorship
No
💼Language
English
🏢Working Hours
Not specified
Apply Now →
You'll be redirected to
the company's application page