GitHub repository: rachthree/tensorflow-hooks
Primary language: Python
Description: PyTorch-like hooks for TensorFlow Keras layers