jax-resnet (pypi): Framework-agnostic library for checking array shapes at runtime.
Keywords: computer-vision|deep-learning|flax|jax|machine-learning|neural-networks|resnet