sparse_autoencoder/requirements.txt

22 lines
287 B
Plaintext

argtoml[save]
build
debugpy
equinox
jax
jaxtyping
jo3mnist
jo3util
matplotlib
optax
pandas
pyright
scikit-learn
tensorflow
torch
torchvision
tqdm
twine
typeguard
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
--extra-index-url https://download.pytorch.org/whl/cpu