761 B
761 B
Simple Transformer
A minimal Transformer implementation using NumPy.
Files
transformer.py- Core Transformer model implementationexample.py- Usage example
Usage
python3 example.py
Model Architecture
The implementation includes:
- Multi-Head Attention: Scaled dot-product attention with multiple heads
- Feed-Forward Network: Two-layer fully connected network with ReLU activation
- Layer Normalization: Applied after each sub-layer
- Positional Encoding: Sinusoidal position embeddings
Model Parameters
Default configuration:
- Vocabulary size: 1000
- Model dimension: 512
- Number of heads: 8
- Number of layers: 6
- Feed-forward dimension: 2048
- Max sequence length: 100
Total parameters: ~19.4M