https://github.com/n2cholas/jax-resnet

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

computer-vision deep-learning flax jax machine-learning neural-networks resnet

Last synced: 16 Jan 2026

https://github.com/n2cholas/shapecheck

Framework-agnostic library for checking array/tensor shapes at runtime.

Last synced: 14 Jan 2026