Foundation-Model Research Engineer Intern

DataRobot, Inc.; Washington (District of Columbia); On-site; USD 40,000 - 70,000; Full time

Benefits offered by this job

Medical, Dental & Vision Insurance

Flexible Time Off Program

Paid Holidays

Paid Parental Leave

Job summary

DataRobot, Inc. is seeking a post-doctoral level researcher/engineer to architect probabilistic foundation models and implement training/evaluation pipelines in production-grade PyTorch code.

The role emphasizes architectural reasoning, rigorous experiments, and scalable software practices to advance state-of-the-art with real-world data while coordinating with teams across DataRobot. This internship targets candidates near PhD level.

Qualifications

• Post-doctoral level or near for architecture of probabilistic foundation models.
• Experience training and evaluating PyTorch models in production-grade code.
• Strong foundation in probability and statistics to reason about densities and dependence.

Responsibilities

• Design and improve the core model architecture for temporal, tabular, and mixed-modality data.
• Run controlled architecture studies, ablations, and memory/throughput profiling.
• Build scalable PyTorch implementations with clear tests and reproducible experiments.

Skills

PyTorch

Transformers

Probabilistic modeling

Stochastic processes

Experimentation

Education

PhD in ML/Statistics/CS

Tools

PyTorch

TensorFlow

Job description

DataRobot, Inc. is seeking a post-doctoral level researcher/engineer to architect probabilistic foundation models and implement training/evaluation pipelines in production-grade PyTorch code.

The role emphasizes architectural reasoning, rigorous experiments, and scalable software practices to advance state-of-the-art with real-world data while coordinating with teams across DataRobot. This internship targets candidates near PhD level.