INSTALLATION
Install Bolt through MLBricks Kit.
TERMINAL
pip install mlbricks-kit==1.0.0b1Import frommlbricks; no separate Bolt package is required for the integrated kit.
REQUIREMENTS
| Python | 3.10+ |
| PyTorch | >=2.10,<2.11 |
| Native CUDA | Optional and used only when a compatible native route is packaged/available. |
VERIFY
PYTHON
import mlbricks
from mlbricks import Bolt, BoltAttention, Attention, BoltModel
print(mlbricks.__version__)
layer = Bolt(d_model=384, num_heads=6)