tfmot.sparsity.keras.prune_low_magnitude

Used in the notebooks

Used in the guide