OpenAI·Machine Learning Engineer·Take-home Assignment
Jun 2026
Took a take-home style ML engineering screen for OpenAI that was basically a full mini-project: build a classifier end-to-end, do the EDA, handle imbalance, tune models, explain your choices, and write it all up. Dense scope for a single assignment.
- Given a labeled dataset, build a complete end-to-end classifier pipeline in Python: perform exploratory data analysis including schema validation, missing value handling, target leakage checks, class imbalance assessment, and feature distributions; create appropriate train/validation/test splits; define a baseline; train at least two models with cross-validation and hyperparameter tuning; handle class imbalance; apply preprocessing; evaluate with business-aligned metrics; run error analysis with feature importances or SHAP; check fairness across subgroups; and deliver a reproducible report with your recommended model.
“This was the whole assignment, not just one question.” The rest of the author's notes on Machine Learning Engineer interview at OpenAI, Take-home Assignment round, covers how they worked through the question, what the panel pushed back on, and what they would do differently.
View Post