Gurumurthy30/Stackformer
Modular PyTorch transformer library for building, training, and experimenting with language and vision models fast.
This is a PyTorch library for machine learning engineers and researchers to quickly build, train, and experiment with advanced language and vision models. It provides modular components to construct transformer architectures like GPT, LLaMA, Gemma, ViT, and SegFormer. You input your datasets and model configurations, and it outputs trained models and performance metrics, streamlining iterative model development.
Available on PyPI.
Use this if you are a machine learning practitioner looking to rapidly prototype and train state-of-the-art transformer models for both text and image data.
Not ideal if you are a business user without a machine learning engineering background, as this is a developer tool requiring coding knowledge.
Stars
7
Forks
—
Language
Python
License
MIT
Category
Last pushed
Mar 12, 2026
Commits (30d)
0
Dependencies
5
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/transformers/Gurumurthy30/Stackformer"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
lucidrains/x-transformers
A concise but complete full-attention transformer with a set of promising experimental features...
kanishkamisra/minicons
Utility for behavioral and representational analyses of Language Models
lucidrains/simple-hierarchical-transformer
Experiments around a simple idea for inducing multiple hierarchical predictive model within a GPT
lucidrains/dreamer4
Implementation of Danijar's latest iteration for his Dreamer line of work
Nicolepcx/Transformers-in-Action
This is the corresponding code for the book Transformers in Action