lightning
torch
torchaudio
torchmetrics
numpy
pandas
wandb
matplotlib
ruff
scikit-learn
vector-quantize-pytorch