AI Researcher — AI Architecture Research jobs in United States
cer-icon
Apply on Employer Site
company-logo

Featherless AI · 13 hours ago

AI Researcher — AI Architecture Research

positionUnited States
timeFull-time
remoteRemote
seniorityMid, Senior Level

Featherlessai is seeking an AI Researcher focused on AI architecture research to help design and advance next-generation model architectures. The role involves publishing novel research while collaborating with engineers to implement ideas into real systems.

Artificial Intelligence (AI)Cloud ComputingDatabase
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

Machine learning researchNeural network architecturesPyTorchJAXPublication record in ML/AISequence modelsAttentionTraining dynamicsOptimizationExperiment designStartup environment adaptabilityCommunication skills

Required

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

Preferred

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

Competitive compensation
Meaningful equity

Company

Featherless AI

twittertwittertwitter
company-logo
We enable serverless inference via our GPU orchestration and model load-balancing system.
dateFounded in 2023
location
San Francisco, California, USA
people-size2-10 employees
web-link
https://featherless.ai/

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 Stage
Total Funding
$5M
Key Investors
Airbus Ventures
2025-10-31Series A
2025-03-17Seed· $5M
Company data provided by crunchbase