Machine Learning Research Engineer (Foundational Research)
Core
Design, code, and evaluate advanced AI systems, specifically Large Language Models (LLMs), focusing on training, post-training techniques, and data-centric research.
Role type
Senior IC Machine Learning Research Engineer (Foundational Research)
Builds
Scalable training infrastructure, data pipelines for synthetic data and curriculum learning, evaluation frameworks, and agentic workflows for LLMs.
Domain
Legal, tax, and accounting data; Large Language Models (LLMs) and Generative AI
Deliverable
production ML models | research
Required skills
PyTorch, Jax, HuggingFace Transformers, vLLM, Ray, Spark, Dask, ClearML, Weights & Biases, MLflow, CI/CD, distributed computing
Preferred skills
Direct Preference Optimization (DPO), Proximal Policy Optimization (PPO), LangGraph, AutoGPT, LlamaIndex, mixed precision training, gradient checkpointing
Technologies
AWS, MS Azure, Google Cloud, Kubernetes (implied by containerization), Docker (implied)
Responsibilities
Build robust systems for training and evaluating large language models; implement novel research ideas in LLM training and agentic systems; develop pipelines for synthetic data generation and automated evaluation; collaborate with researchers to translate papers into scalable implementations.
Seniority
Senior, hands-on IC
