Causal tracing for language models
MIT License
Home of StarCoder: fine-tuning & inference!
Neural network visualization toolkit for tf.keras
All Model summary in PyTorch similar to `model.summary()` in Keras
View model summaries in PyTorch!
Ongoing research training transformer language models at scale, including: BERT & GPT-2
torchview: visualize pytorch models
OCR-D wrapper for detectron2 based segmentation models
Unsupervised Language Modeling at scale for robust sentiment classification
A comparison and discussion of different NLP methods for 5-class sentiment classification on the ...
Python profiler for pytorch models. Wraps each layer and measures it's execution time.
Visual Attention based OCR
Generative Adversarial Transformers