Train ML models on Databricks. Use for: classification/regression/deep-learning (XGBoost, scikit-learn, LightGBM, PyTorch) with Optuna, @prod/@challenger aliases, batch scoring (spark_udf for plain models, fe.score_batch for feature-store-backed), custom PyFunc, custom ResponsesAgent (LangGraph + UC Function/Vector Search); UC feature tables + FeatureLookup + point-in-time joins + Lakebase…
Install
npx skills add databricks/databricks-agent-skills --skill databricks-ml-trainingnpx skills add databricks/databricks-agent-skills --skill databricks-ml-training -a claude-code -gnpx skills add databricks/databricks-agent-skills --skill databricks-ml-training -a codex -ggemini skills install https://github.com/databricks/databricks-agent-skills.git --path skills/databricks-ml-traininggh skill install databricks/databricks-agent-skills databricks-ml-traininggit clone --depth 1 https://github.com/databricks/databricks-agent-skills /tmp/databricks__databricks-agent-skills
mkdir -p ~/.claude/skills ~/.agents/skills
cp -R /tmp/databricks__databricks-agent-skills/skills/databricks-ml-training ~/.claude/skills/databricks-ml-training # Claude Code
cp -R /tmp/databricks__databricks-agent-skills/skills/databricks-ml-training ~/.agents/skills/databricks-ml-training # Codex, Cursor, Gemini CLI, Copilot, OpenCode