← all repositories

lucidrains/x-transformers

A full-featured PyTorch transformer library implementing encoder-decoder, GPT-style decoder-only, and BERT-style encoder-only architectures.

5.9k stars Python ML Frameworks
x-transformers
Velocity · 7d
+2.9
★ / day
Trend
steady
star history

This repository provides a complete transformer implementation in PyTorch with support for encoder-only, decoder-only, and encoder-decoder architectures. It includes experimental features from various research papers, enabling users to build BERT-style, GPT-style, and full encoder-decoder models with configurable dimensions, depth, attention heads, and sequence lengths. The library is installable via pip and designed for researchers and practitioners implementing transformer-based models.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.