lightning
transformers
diffusers
accelerate
toml
timm
pytorch-nlp
datasets
onnx
onnxruntime
optimum
black
GitPython
colorlog
cocotb[bus]==1.8.0
pytest
pytest-cov
pytest-xdist
pytest-sugar
pytest-html
pytest-profiling
ipython
ipdb
sentencepiece
einops
deepspeed
pybind11
tabulate
tensorboard
optuna
stable-baselines3
scikit-learn
h5py
pyyaml
numpy
pandas
wandb
imageio
imageio-ffmpeg
opencv-python
kornia
einops
sphinx
sphinx-rtd-theme
