PyTorch Module
PyTorch Generator
Generate a ready-to-run PyTorch project in seconds.
11 flags 7 files
Template Library
Pick a blueprint and let the config snap it into place.
Hugging Face Blueprint
Hugging Face Text Classification
Fine-tune BERT/RoBERTa/DistilBERT/etc. for text classification using Hugging Face Trainer, Datasets, and evaluation.
5 flags 8 files
Coming soon
Hugging Face Vision Blueprint
Hugging Face Image Classification
Fine-tune ViT, Swin Transformer, ConvNeXt, or similar for image classification with Hugging Face ecosystem.
5 flags 8 files
Coming soon
LLM Blueprint
NanoGPT-style LLM generator
Minimal GPT-like transformer trained from scratch (nanoGPT inspired): tokenizer, model definition, data prep, and training loop.
6 flags 9 files
Coming soon
Time-Series Blueprint
Time-Series Forecasting Models
LSTM / GRU / Transformer / TCN for univariate/multivariate time-series forecasting with sliding windows and metrics (MAE, RMSE).
5 flags 8 files
Coming soon
Distributed Blueprint
PyTorch Distributed Data Parallel (DDP)
Multi-GPU / multi-node training with torch.distributed + DDP, launcher support, syncBN option, and logging.
5 flags 7 files
Coming soon
Hugging Face Finetuning Blueprint
General Hugging Face Finetuning
Flexible Trainer-based finetuning for any task (text, vision, multimodal, custom head) with PEFT/LoRA support, custom datasets, and eval.
6 flags 10 files
Coming soon