AI Researcher - AI Architecture Research
WorldwideFull-TimeMiddle
Salary not disclosed
Apply NowOpens the employer's application page
Job Details
- Required Skills
- Machine LearningPyTorch
Requirements
- Strong background in machine learning research, with a focus on model architecture
- Publication record in ML/AI venues (e.g. NeurIPS, ICML, ICLR, COLM, ACL, EMNLP, arXiv)
- Deep understanding of neural network architectures
- Deep understanding of sequence models and attention mechanisms
- Deep understanding of training dynamics and optimization
- Hands-on experience with PyTorch or JAX
- Ability to reason rigorously, design clean experiments, and communicate results clearly
- Comfortable working in a fast-moving startup environment
Responsibilities
- Research and design novel AI architectures (e.g. alternatives to standard Transformer designs, long-context models, efficient sequence modeling, hybrid architectures)
- Explore architectural improvements for scalability, efficiency, and stability
- Prototype and evaluate new architectures through ablations, benchmarks, and empirical studies
- Author and co-author research papers for top ML conferences and journals
- Collaborate with engineering teams to translate research into training and inference systems
- Stay current with state-of-the-art research and identify promising directions early
View Full Description & ApplyYou'll be redirected to the employer's site
