Featherless AI · 12 hours ago
AI Researcher — Training Optimization
United States
Full-time
Remote
Mid, Senior LevelFeatherlessAI is seeking an AI Researcher focused on training optimization to enhance the efficiency and scalability of large-scale model training. The role involves developing innovative techniques for training optimization and conducting rigorous experiments to validate findings.
Artificial Intelligence (AI)Cloud ComputingDatabase
Responsibilities
Design and evaluate training optimization techniques for large models (e.g. optimization algorithms, schedulers, normalization, curriculum strategies)
Improve training efficiency and stability across long runs and large datasets
Research and implement methods such as:
Optimizer and scheduler innovations
Mixed-precision, low-precision, and memory-efficient training
Gradient noise reduction, scaling laws, and convergence analysis
Training-time regularization and robustness techniques
Run large-scale experiments, analyze results, and translate findings into actionable improvements
Author or co-author research papers, technical reports, or blog posts
Collaborate closely with infrastructure and inference teams to ensure training decisions translate to real-world performance
Qualification
Required
Strong background in machine learning research, with emphasis on training dynamics and optimization
Experience training large neural networks (LLMs, multimodal models, or large sequence models)
Publication experience in ML venues (e.g. NeurIPS, ICML, ICLR, ACL, EMNLP, COLM, arXiv) or equivalent high-quality open research
Solid understanding of optimization theory and practice
Solid understanding of backpropagation, gradient flow, and training stability
Solid understanding of distributed and large-batch training
Proficiency in Python and modern ML frameworks (PyTorch preferred)
Ability to independently design experiments and reason from data
Preferred
Experience with non-standard architectures (e.g. RNN variants, long-context models, hybrid systems)
Experience optimizing training on GPUs at scale (FSDP, ZeRO, custom kernels)
Contributions to open-source ML or research codebases
Comfort operating in fast-moving, ambiguous startup environments
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