seralexger/jax-image-classifier

Image classifier with Jax and Elegy. Using as an input any net to get the bottleneck features.

Jupyter Notebook