Skip to content

[Flax] adding support for batch norm layers - #21581

Merged
sgugger merged 7 commits into
huggingface:mainfrom
Shubhamai:flax-batchnorm-support
Feb 24, 2023
Merged

[Flax] adding support for batch norm layers#21581
sgugger merged 7 commits into
huggingface:mainfrom
Shubhamai:flax-batchnorm-support

simplifying checking batch norm layer

1c2e91a
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs