AI Research Scientist

About the role

We are looking for an AI Research Scientist to help design, train, evaluate, and improve advanced AI systems. This role sits at the intersection of deep learning research, applied machine learning, and scalable engineering.

You will work on research problems involving large language models, multimodal reasoning, synthetic data generation, model evaluation, representation learning, and task-specific adaptation. You will be expected to move from ambiguous research questions to concrete hypotheses, experiments, prototypes, and eventually production-ready methods in collaboration with engineering and product teams.

This is a hands-on research role: you will read papers, design experiments, train and fine-tune models, build evaluation pipelines, analyze failures, and help translate research insights into reliable AI capabilities.

As an AI Research Scientist, you will:

• Define and execute research projects around LLMs, reasoning, multimodal models, synthetic data, and model adaptation.
• Design experiments to test hypotheses, compare architectures, evaluate training strategies, and measure model behavior.
• Train, fine-tune, and evaluate Transformer-based models using modern deep learning frameworks.
• Work on supervised fine-tuning, preference optimization, LoRA/PEFT methods, distillation, data augmentation, and evaluation-driven model improvement.
• Build robust benchmarks and diagnostic evaluations for reasoning, generalization, reliability, and task-specific performance.
• Analyze model failures and propose improvements at the level of data, architecture, training objective, prompting, or inference strategy.
• Collaborate with ML engineers to scale experiments across GPUs and distributed infrastructure.
• Contribute clean, reproducible research code, experiment configs, documentation, and internal reports.
• Stay up to date with relevant AI research and translate promising ideas into practical experiments.
• Help shape the company’s research roadmap and identify high-impact technical directions.

We are looking for someone with strong experience in several of the following areas:

• Deep learning, especially Transformer architectures and modern sequence models.
• LLM training, fine-tuning, evaluation, or inference.
• Strong practical experience with Python and PyTorch.
• Experience with the Hugging Face ecosystem: transformers, datasets, tokenizers, model checkpoints, and generation APIs.
• Understanding of training dynamics, optimization, loss functions, overfitting, regularization, and evaluation methodology.
• Ability to design rigorous experiments and interpret results beyond headline metrics.
• Experience working with large datasets, preprocessing pipelines, and reproducible ML workflows.
• Strong mathematical foundations in linear algebra, probability, statistics, and optimization.
• Ability to read research papers and turn them into working prototypes.
• Clear communication skills and the ability to explain research tradeoffs to technical and non-technical stakeholders.

The role should stay close to the current ML engineering stack while adding research-oriented tools.

Expected core stack:

• Python
• Py Torch
• Hugging Face Transformers / Datasets
• CUDA-aware GPU training
• MLflow or Weights & Biases for experiment tracking
• Docker
• Git Lab CI
• GCP / cloud GPU infrastructure
• Ray or similar tools for distributed workloads and experiment orchestration
• PyTorch Distributed, FSDP, Deep Speed, or Accelerate
• PEFT / LoRA / QLoRA
• vLLM
• pytest and reproducibility tooling for research code quality

Nice-to-have tooling:

• Triton or custom CUDA kernels
• GCS
• RAG pipelines, vector databases, and embedding evaluation
• Data annotation, synthetic data generation, and human-evaluation pipelines