KFAC impelementation for batchnorm

Is there an implementation for kfac that can handle batchnorm? I found this implementation of kfac but this does not support batchnorm. I need some hint on how to extend it to batchnorm. I greatly appreciate it.