flax

509780a1 - Merge pull request #5174 from ayulockedin:doc/audit-conv-layers