Machine Learning Engineer — AI Architecture Research
New
J
JobgetherArtificial Intelligence
USFull-Time
Salary not disclosed
Apply NowOpens the employer's application page
Job Details
- Required Skills
- Machine LearningPyTorchDeep Learning
Requirements
- Strong foundation in machine learning and deep learning fundamentals, with practical experience applying them to model development.
- Hands-on experience implementing neural network or model architectures from scratch.
- Strong understanding of attention mechanisms, RNNs, state-space models, hybrid architectures, or related approaches.
- Solid knowledge of training dynamics, optimization, scaling behavior, and architecture-level performance considerations.
- Understanding of model-level memory, latency, compute, and efficiency constraints.
- Proficiency with PyTorch or JAX and the ability to develop and experiment with research-oriented ML code.
- Ability to evaluate architectural ideas through both theoretical reasoning and empirical experimentation.
- Strong communication skills, with the ability to clearly explain technical concepts and architectural trade-offs.
Responsibilities
- Research and develop novel neural network architectures, including alternatives or extensions to Transformers, recurrent and hybrid models, and long-context systems.
- Design and execute architecture-level experiments focused on scaling laws, memory mechanisms, training behavior, and compute-performance trade-offs.
- Prototype models end-to-end, translating research concepts into robust, training-ready implementations.
- Analyze model behavior, failure modes, inductive biases, and architectural strengths and limitations.
- Collaborate with inference and systems engineering teams to ensure new architectures are efficient, scalable, and suitable for deployment.
- Read, reproduce, evaluate, and extend cutting-edge machine learning research papers.
- Contribute to internal research notes, benchmarks, experiments, and open-source initiatives where applicable.
- Move fluidly between theoretical investigation, rapid experimentation, and production-oriented engineering.
View Full Description & ApplyYou'll be redirected to the employer's site
