Pre-built ML models you can train and benchmark on the tracebloc platform — or use as a starting point for your own.
Every model in this repo is compatible with tracebloc's secure training environment. Pick one, upload it, and start an experiment in minutes. No setup, no configuration, no boilerplate.
| Task | Framework | Path | Use when you need to… |
|---|---|---|---|
| Image classification | PyTorch, TensorFlow | model_zoo/image_classification/ |
Categorize images into labels |
| Object detection | PyTorch | model_zoo/object_detection/pytorch/ |
Locate and classify objects in images |
| Text classification | PyTorch | model_zoo/text_classification/pytorch/ |
Classify documents, reviews, tickets |
| Token classification | PyTorch | model_zoo/token_classification/pytorch/ |
Tag tokens: NER, POS, slot filling |
| Causal language modeling | PyTorch | model_zoo/causal_language_modeling/pytorch/ |
Pretrain/fine-tune a decoder-only LM (next-token) |
| Masked language modeling | PyTorch | model_zoo/masked_language_modeling/pytorch/ |
Pretrain/domain-adapt an encoder LM (fill-in-the-blank) |
| Seq2seq | PyTorch | model_zoo/seq2seq/pytorch/ |
Map text to text: translation, summarization |
| Embeddings | PyTorch | model_zoo/embeddings/pytorch/ |
Learn text embeddings for search, retrieval, similarity |
| Sentence pair classification | PyTorch | model_zoo/sentence_pair_classification/pytorch/ |
Score sentence pairs: NLI, duplicates, relevance |
| Semantic segmentation | PyTorch | model_zoo/semantic_segmentation/pytorch/ |
Pixel-level image labeling |
| Keypoint detection | PyTorch | model_zoo/keypoint_detection/pytorch/ |
Detect landmarks on objects or bodies |
| Tabular classification | PyTorch, sklearn, TensorFlow | model_zoo/tabular_classification/ |
Classify structured/tabular data |
| Tabular regression | PyTorch, sklearn, TensorFlow | model_zoo/tabular_regression/ |
Predict continuous values from tables |
| Time series forecasting | PyTorch | model_zoo/time_series_forecasting/pytorch/ |
Forecast future values from sequences |
| Time-to-event prediction | lifelines, PyTorch, scikit-survival | model_zoo/time_to_event_prediction/ |
Predict when an event will occur |
git clone https://github.com/tracebloc/model-zoo.gitThen upload a model to tracebloc:
from tracebloc_package import User
user = User() # log in with your tracebloc email + password
user.uploadModel("model-zoo/model_zoo/image_classification/tensorflow/densenet.py")Full walkthrough → Open the training notebook in Colab
Not limited to this zoo. Any model that follows the model structure requirements works — PyTorch, TensorFlow, or custom containers.
Models in this zoo do not ship pretrained weights. Each model loads weights through its underlying library at runtime — torchvision's weights="DEFAULT", HuggingFace's from_pretrained(), or training from scratch.
If you're uploading your own model and want to bundle pretrained weights with it, name the weights file mymodel_weights.pkl and place it in the same directory as mymodel.py. The platform will pick it up automatically.
Platform · Docs · Training notebook · PyPI package · Discord
Apache 2.0 — see LICENSE.
Questions? support@tracebloc.io or open an issue.