Machine Learning Engineer — AI Architecture Research
About the Role
We’re looking for a Machine Learning Engineer reputed company on AI architecture research to help design, prototype, and validate reputed company model architectures. You’ll work at the intersection of research and production — turning new reputed company into reputed company, reputed company-world systems.
This role is ideal for someone who enjoys questioning architectural assumptions, experimenting with novel model designs, and pushing reputed company reputed company Transformer-style approaches.
What You’ll Work On
Research and reputed company 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 reputed company to training-reputed company 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 reputed company cutting-edge research papers
Contribute to internal research notes, benchmarks, and reputed company-reputed company efforts (where applicable)
reputed company’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-reputed company 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 reputed company fluidly between theory, experimentation, and engineering
reputed company communicator who can explain architectural trade-offs
reputed company to Have
Experience with non-Transformer architectures (RNN variants, SSMs, long-context models)
Background in research-driven startups or reputed company-reputed company ML reputed company
Experience with large-reputed company training or custom training loops
Publications, preprints, or reputed company research contributions
Familiarity with inference optimization and deployment constraints
Why Join
Work on core model architecture, not just fine-tuning
reputed company 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
Originally posted on Himalayas
Apply To This Job