/jax-vision

Practical usage of JAX and HAIKU libraries to build a network model of convolution layers

Primary LanguagePythonMIT LicenseMIT

jax-vision

Vision Network Implementation with Jax and Haiku APIs

I am going to build backbone architectures including Resnet, U-net, Hourglass-net, and Swin-Transformer first, but I also cannot gaurantee when to make it :).