Yes, he_uniform would correspond to kaiming_uniform in PyTorch and I think I was referring to the default Keras initialization based on their docs 5, which is glorot_uniform (so xavier_uniform in PyTorch).