Machine Learning Engineer — AI Architecture Research
Featherless AI4 months ago
Remote, WorldwideMid Level / Senior
Responsibilities
- Research and develop new neural network architectures.
- Design and run architecture-level experiments.
- Prototype models end-to-end from research code to training-ready implementations.
- Collaborate with inference and systems engineers for deployable architectures.
- Analyze model behavior, failure modes, and inductive biases.
- Read, reproduce, and extend cutting-edge research papers.
- Contribute to internal research notes, benchmarks, and open-source efforts.
Requirements
- Strong background in machine learning fundamentals and deep learning.
- Hands-on experience implementing model architectures from scratch.
- Solid understanding of attention mechanisms, RNNs, and hybrid architectures.
- Comfortable working in PyTorch or JAX.
- Ability to move fluidly between theory, experimentation, and engineering.
- Clear communicator who can explain architectural trade-offs.
Benefits
- Work on core model architecture, not just fine-tuning.
- Direct influence on the technical direction of a Series-A company.
- Small, high-caliber team with fast feedback loops.
- Opportunity to ship research into production.
- Competitive compensation + meaningful equity.
Tech Stack
PyTorch
Categories
AI & MLData Science