swyang50066 / jax-vision

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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 :).

About

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

License:MIT License


Languages

Language:Python 100.0%