---
title: 'Machine Learning Engineer — Training Optimization at Featherless AI'
canonical: 'https://feeny.ai/job/machine-learning-engineer-training-optimization-featherless-ai-world-xs6g96apkhzx'
type: 'job'
last_seen: '2026-09-08'
---

# Machine Learning Engineer — Training Optimization at Featherless AI

- **Company:** Featherless AI
- **Location:** World
- **Employment:** full-time
- **Work type:** remote
- **Posted:** 2026-01-22
- **Last confirmed live:** 2026-09-08
- **Apply:** https://jobs.ashbyhq.com/featherlessai/dcbb7273-4d0e-4fd0-bb33-1e0ad99eca69

## Job description

## ABOUT THE ROLE

We’re looking for an ML Engineer focused on training optimization to help us scale and improve large-scale model training. You’ll work at the intersection of research and production, optimizing training pipelines for speed, stability, and cost—while collaborating closely with researchers pushing model architecture and capability forward.

This is a high-impact role with real ownership: your work directly affects how fast we can iterate, how large we can scale, and how efficiently we deploy new models.

## WHAT YOU’LL DO

- Optimize large-scale model training pipelines (throughput, convergence, stability, and cost)
- Improve distributed training strategies (data, model, and pipeline parallelism)
- Tune optimizers, schedulers, batch sizing, and precision (bf16 / fp16 / fp8)
- Reduce training time and compute cost via profiling, bottleneck analysis, and systems-level improvements
- Collaborate with researchers on architecture-aware training strategies
- Build and maintain robust training infrastructure (checkpointing, fault tolerance, reproducibility)
- Evaluate and integrate new training techniques (e.g. gradient checkpointing, ZeRO, FSDP, custom kernels)
- Own training performance metrics and continuously push them forward

## WHAT WE’RE LOOKING FOR

- Strong experience training large neural networks (LLMs or similarly large models)
- Hands-on experience with training optimization (not just model usage)
- Solid understanding of:
- Backpropagation, optimization algorithms, and training dynamics
- Distributed systems for ML training
- Experience with PyTorch (required)
- Comfort working close to hardware (GPUs, memory, networking constraints)
- Ability to move fluidly between research ideas and production-ready code

## NICE TO HAVE

- Experience with large-scale distributed training (multi-node, multi-GPU)
- Familiarity with DeepSpeed, FSDP, Megatron, or custom training stacks
- Experience optimizing training on AMD or NVIDIA GPUs
- Contributions to open-source ML infrastructure or research codebases
- Exposure to non-Transformer architectures (RNNs, hybrid models, etc.)

## WHY JOIN US

- Real ownership at Series-A stage — your work shapes the company’s trajectory
- Work on cutting-edge models and training systems at scale
- Small, highly technical team with fast feedback loops
- Strong emphasis on engineering quality and research rigor
- Competitive compensation + meaningful equity

## About Featherless AI

## Company Overview
- **One-liner**: Featherless AI provides a serverless platform that offers API access to over 40,000 open-weight AI models from a single endpoint, designed for developers and enterprises.
- **Entity Type**: Private (Series A)
- **Headquarters**: San Francisco, California, United States
- **Founded**: 2023
- **Founders**: Eugene Cheah (CEO, Co-Founder)

## Core Business
- **Primary industry**: Artificial Intelligence Infrastructure / Serverless LLM Hosting
- **Target customers**: B2B, serving developers, AI startups, and enterprises seeking scalable, cost-effective inference for open-source models.
- **Mission or purpose**: To democratize access to all AI models by making them available for serverless inference, eliminating the need for server setup and complex infrastructure management.

## Products & Services
- **Featherless API**: A unified API gateway providing instant access to over 40,000 open-weight models (e.g., DeepSeek, Llama, Mistral, Qwen, RWKV, GLM, Kimi) without setup or hosting. Pricing is flat-rate with unlimited tokens, starting at $25/month for up to 4 concurrent connections and 32K context, scaling to $200/month for higher tiers. Agent-specific plans ($100/month) include sandbox environments and persistent storage. The service emphasizes low latency, dependable uptime, and predictable costs.

## Market Standing
- **Valuation/Market Cap**: Not publicly disclosed.
- **Key Metric**: **Total Funding of $25M** — raised $5M in a Seed round (April 2025, led by Airbus Ventures) and $20M in a Series A round (announced ~May 2026, details still emerging).
- **Notable Investors/Partners**: Airbus Ventures, Kickstart Ventures, Panache Ventures, BMW i Ventures, AMD Ventures, and 11 other investors. The platform is built by researchers contributing to RWKV, a Linux Foundation project.
- **Growth Signals**: The company is on a rapid growth trajectory, with headcount increasing 46.7% year-over-year to 14 employees. The website boasts over 2,100 "stars" for its Discord community. The company has a global presence, operating in 9 countries (including Singapore, Canada, Czechia, UK, Belgium, and Sweden). Website traffic is strong (73,508 monthly visits, growing +19.9% month-over-month), and there are 43 active job postings, a 65.4% quarterly increase in hiring. The Series A announcement signals significant investor confidence.

## Competitive Advantages
- **Extensive Model Library & Zero-Friction Access**: A single API key provides access to the entire Hugging Face trending library, including models up to 229B parameters, with no need to manage infrastructure.
- **Unlimited-Token, Flat-Rate Pricing**: A strong differentiator in the "per-token" pricing era, offering predictable costs suitable for scaling, with tiers from $25 to $200/month.
- **Build for Reliability & Performance**: Architecture designed for real workloads with low latency and dependable uptime, utilizing proprietary GPU orchestration and model load-balancing.
- **Open-Source Roots & Community**: Built by researchers contributing to RWKV (a Linux Foundation project), the company is deeply embedded in the open-source AI ecosystem, which fosters trust and community-driven development.

## Strategic Focus
- **Scaling the Platform & Enterprise Adoption**: The Series A funding will be used to expand AI infrastructure and grow the platform. The company is actively hiring for senior roles like Founding Account Executives and Business Development Reps, signaling a shift toward aggressive go-to-market and enterprise sales.
- **Expanding Global Presence**: With a distributed team across the US, Europe, and Asia, the company is building a global, remote-first workforce.
- **Technical Innovation**: Focused on continuous improvement of inference performance and cost-efficiency through their GPU orchestration system and model load-balancing.

## Why Work Here
- **High-Growth Stage**: As a Series A startup with strong investor backing, this is an opportunity to join a company experiencing rapid scaling, which offers significant career growth and impact potential.
- **Impact & Ownership**: Employees are likely to have high autonomy and a direct impact on the company's trajectory, from building core infrastructure to driving revenue.
- **Remote-First & Global Team**: Based on the distributed headcount across 9 countries (US, Singapore, Canada, UK, Belgium, etc.), the company is clearly remote-first, offering flexibility in where you work. Job postings reflect opportunities in the US and Europe (e.g., Paris, Berlin).
- **Cutting-Edge Technical Challenge**: The core work involves solving complex problems in AI inference, GPU orchestration, and MLOps, making it a compelling place for engineers and researchers passionate about AI infrastructure.
- **Culture & Values**: The company's deep ties to open-source AI communities and its "flat-rate, no-surprises" pricing philosophy likely translate into a transparent, developer-friendly internal culture. The small, highly-skilled team (14 people) suggests a close-knit, high-performing environment.

## Sources
1. [Featherless.ai Website](https://featherless.ai/)
2. [Featherless AI LinkedIn](https://www.linkedin.com/company/feather-serverless-ai)
3. [Featherless AI Docs](https://featherless.ai/docs/overview)
4. [CB Insights Profile](https://www.cbinsights.com/company/recursal-ai)
5. [Featherless AI Jobs](https://jobs.ashbyhq.com/featherlessai)

## Other roles at Featherless AI

- [Founding Account Executive (AI Cloud)](https://feeny.ai/job/founding-account-executive-ai-cloud-featherless-ai-us-krryw5zpmt01) — US &, Canada
- [Founding Business Development Rep (AI Cloud US/CA)](https://feeny.ai/job/founding-business-development-rep-ai-cloud-us-ca-featherless-ai-us-y876z86dw2y3) — US &, Canada
- [Chief of Staff](https://feeny.ai/job/chief-of-staff-featherless-ai-san-francisco-15ydsmveqwsq) — San Francisco, CA
- [Content Marketer](https://feeny.ai/job/content-marketer-featherless-ai-europe-xnsbz1evvkbs) — Europe
- [Business Development Rep (AI Cloud)](https://feeny.ai/job/business-development-rep-ai-cloud-featherless-ai-europe-hd5fymsqsh5h) — Europe
- [AI Researcher — Training Optimization](https://feeny.ai/job/ai-researcher-training-optimization-featherless-ai-world-hyc2csn67p2p) — World
- [AI Researcher – Multilingual Data](https://feeny.ai/job/ai-researcher-multilingual-data-featherless-ai-world-aj24t2jw442p) — World
- [AI Researcher — AI Architecture Research](https://feeny.ai/job/ai-researcher-ai-architecture-research-featherless-ai-world-dprg8203nt10) — World
- [AI Researcher — Distillation](https://feeny.ai/job/ai-researcher-distillation-featherless-ai-world-zxs1tx4mwq1f) — World
- [AI Researcher — Inference Optimization](https://feeny.ai/job/ai-researcher-inference-optimization-featherless-ai-world-5q35t4rfgjve) — World