Machine Learning Engineer — AI Architecture Research

About this role

ABOUT THE ROLE

We’re looking for a Machine Learning Engineer focused on AI architecture research to help design, prototype, and validate next-generation model architectures. You’ll work at the intersection of research and production — turning new ideas into scalable, real-world systems.

This role is ideal for someone who enjoys questioning architectural assumptions, experimenting with novel model designs, and pushing beyond standard Transformer-style approaches.

WHAT YOU’LL WORK ON

- Research and develop new neural network architectures (e.g. alternatives or extensions to Transformers, recurrent / hybrid models, long-context systems)

- Design and run architecture-level experiments (scaling laws, memory mechanisms, compute trade-offs)

- Prototype models end-to-end — from research code to training-ready implementations

- Collaborate with inference and systems engineers to ensure architectures are deployable and efficient

- 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 (where applicable)

WHAT WE’RE LOOKING FOR

- Strong background in machine learning fundamentals and deep learning

- Hands-on experience implementing model architectures from scratch

- Solid understanding of: - Attention mechanisms, RNNs, state-space models, or hybrid architectures - Training dynamics, scaling behavior, and optimization - Memory, latency, and compute constraints at the model level

- Comfortable working in PyTorch or JAX

- Ability to move fluidly between theory, experimentation, and engineering

- Clear communicator who can explain architectural trade-offs

NICE TO HAVE

- Experience with non-Transformer architectures (RNN variants, SSMs, long-context models)

- Background in research-driven startups or open-source ML projects

- Experience with large-scale training or custom training loops

- Publications, preprints, or notable research contributions

- Familiarity with inference optimization and deployment constraints

WHY JOIN

- 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

Similar jobs

• AI Researcher — AI Architecture Research — Featherlessai · Remote
• Machine Learning Research Engineer — Andurilindustries · Washington, District of Columbia, United States
• Sr Machine Learning Engineer, AI Research — Cribl · Remote
• Machine Learning Engineer, AI Labs — Netskope · Santa Clara, California, United States
• Principal AI/ML Architect — Caylent · Remote
• Applied AI Research Engineer — Netic · San Francisco
• Senior AI/ML Architect, Applied Field Engineering — Snowflake · US-NY-New York
• Principal AI/ML Architect, Applied Field Engineering — Snowflake · US-NY-New York