AI Researcher — AI Architecture Research

The role

You will research and design novel AI architectures. You will prototype and evaluate new architectures and collaborate with engineers to translate research into training and inference systems.

What you will do

• Research and design novel AI 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

Required qualifications

• 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
• Deep understanding of sequence models and attention mechanisms
• Deep understanding of 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

Benefits

• 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

About the company

featherlessai is a small, highly technical team with a strong research culture.