attn-gym
bitarray==3.0.0
deflate
lightning[pytorch-extra]
numba>=0.59.1
tqdm
torch>=2.5.0
