Remote Machine Learning Engineer — AI Architecture Research

Description:

• Seeking a Machine Learning Engineer focused on AI architecture research to design, prototype, and validate next-generation model architectures.
• Role involves turning new ideas into scalable, real-world systems, questioning architectural assumptions, and experimenting with novel model designs.

Requirements:

• 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, training dynamics, scaling behavior, and optimization.
• 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.