featherlessai - AI Researcher — AI Architecture Research

Remote (world) - Hybrid+ Equity9mo ago

In Office WW Artificial Intelligence Architecture Research Engineer JAX

Requirements

• Strong background in machine learning research, with a focus on model architecture • machine learning research • Publication record in ML/AI venues (e.g. NeurIPS, ICML, ICLR, COLM, ACL, EMNLP, arXiv) • Publication record • Deep understanding of: • Neural network architectures • Sequence models and attention mechanisms • Training dynamics and optimization • Ability to reason rigorously, design clean experiments, and communicate results clearly • Comfortable working in a fast-moving startup environment • Experience with non-Transformer architectures (e.g. RNN-based, state-space, hybrid models) • non-Transformer architectures • 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

Responsibilities

• 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.

Benefits

• High ownership over research direction and roadmap • Clear path to publishing impactful work • 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