AI Research Intern

About the role

Lexi Labs is hiring AI Research Interns for Bangalore or remote work. The role focuses on machine learning research, transformer architectures, model training and evaluation, interpretability, optimisation, and LLM alignment.

Responsibilities

• Build clean, modular, and testable Python code.
• Work on machine learning and deep learning models.
• Explore transformer architectures and model training techniques.
• Contribute to model evaluation, interpretability, and optimisation.
• Work with version control, packaging, documentation, and collaborative development practices.

Requirements

• Strong Python expertise.
• Deep understanding of machine learning and deep learning principles, with hands-on PyTorch experience.
• Comprehensive knowledge of attention mechanisms, positional encodings, tokenisation, and training objectives in architectures such as BERT, GPT, LLaMA, T5, and Mamba.
• Knowledge of Git workflows, packaging, documentation, and collaborative development practices.
• Excellent communication, peer code review, and teamwork skills.

Required Skills