2021-07-06 17:11:35 +00:00
|
|
|
# Fundamentals
|
2019-11-24 07:48:52 +00:00
|
|
|
numpy
|
|
|
|
pyyaml
|
|
|
|
tb-nightly
|
|
|
|
future
|
2020-06-01 21:09:52 +00:00
|
|
|
scp
|
|
|
|
tqdm
|
2020-08-03 22:57:56 +00:00
|
|
|
matplotlib
|
|
|
|
scipy
|
2020-09-08 14:04:16 +00:00
|
|
|
munch
|
2020-09-09 16:46:08 +00:00
|
|
|
tqdm
|
|
|
|
scp
|
2020-10-27 21:24:05 +00:00
|
|
|
tensorboard
|
2021-01-12 03:09:16 +00:00
|
|
|
orjson
|
2021-03-03 03:51:48 +00:00
|
|
|
einops
|
2021-06-05 05:23:20 +00:00
|
|
|
lambda-networks
|
2022-06-10 03:15:09 +00:00
|
|
|
mup
|
2021-07-06 17:11:35 +00:00
|
|
|
|
|
|
|
# For image generation stuff
|
|
|
|
opencv-python
|
|
|
|
kornia
|
|
|
|
pytorch_ssim
|
|
|
|
gsa-pytorch
|
|
|
|
pytorch_fid==0.1.1
|
|
|
|
|
|
|
|
# For audio generation stuff
|
|
|
|
inflect==0.2.5
|
|
|
|
librosa==0.6.0
|
2021-07-20 14:36:46 +00:00
|
|
|
Unidecode==1.0.22
|
|
|
|
tgt == 1.4.4
|
2021-08-11 23:32:06 +00:00
|
|
|
pyworld == 0.2.10
|
2021-12-22 20:44:02 +00:00
|
|
|
audio2numpy
|
|
|
|
|
|
|
|
# For text stuff
|
|
|
|
transformers
|
|
|
|
tokenizers
|
2022-01-16 00:28:59 +00:00
|
|
|
jiwer # calculating WER
|
2021-12-22 20:44:02 +00:00
|
|
|
|
|
|
|
# lucidrains stuff
|
|
|
|
vector_quantize_pytorch
|
|
|
|
linear_attention_transformer
|
|
|
|
rotary-embedding-torch
|
|
|
|
axial_positional_embedding
|
2022-01-11 23:25:40 +00:00
|
|
|
g-mlp-pytorch
|
|
|
|
x-clip
|