A High Level API for Deep Learning in JAX
Why do you think that https://github.com/n2cholas/jax-resnet is a good alternative to elegy
A High Level API for Deep Learning in JAX
Why do you think that https://github.com/n2cholas/jax-resnet is a good alternative to elegy