search

kmheckel / spyx

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

Refactor dataloading to not require torchvision. #7

Closed kmheckel closed 7 months ago

kmheckel commented 11 months ago

Right now the dataloading part of the library uses torch dataloaders and torchvision. This increases the dependency requirements and makes installing less convenient, so it would be nice to use a more JAX native approach if possible.

This might be a potential solution:

https://github.com/birkhoffg/jax-dataloader

Ideally, make it that spyx.data is only installed explicitly as "pip install spyx[data]"

kmheckel commented 7 months ago

As of v0.1.17 the prebuilt dataloaders have been moved to optional submodule spyx.loaders