Build deep learning code ready to train and test.
Guides agents in writing modular deep learning code with efficient training, evaluation, and interactive demos.
When to use it
Use when writing Python deep learning code for transformers, diffusion models, or LLMs.
It changes how the agent writes deep learning project code, emphasizing modular structure, configuration, and experiment tracking.
What you provide
Lists torch as a required dependency and specifies PyTorch as the primary framework.
Lists transformers as a required dependency.
Lists diffusers as a required dependency.
Lists gradio as a required dependency.
No additional actions listed.