search

kmheckel / spyx

Spyx: Spiking Neural Networks in JAX
https://spyx.readthedocs.io/en/latest/
MIT License
98 stars 11 forks source link

Find way to do dataloading without torchvision #22

Closed kmheckel closed 6 months ago

kmheckel commented 6 months ago

I would love to not need to install torchvision and PyTorch just to load data into Spyx, as Torch often lags JAX in terms of CUDA support.

kmheckel commented 6 months ago

Torchvision has been eliminated as a dependency, so users are not forced to install it and PyTorch if they don't wish to do so. Users could use Tonic and then Tensorflow datasets as a possible alternative.