4 ms·
Nice! The README mentions `LayerNorm` is implemented here, but while it's in the equivalence tests with PyTorch, I don't see it in the implementation.
by gfaure 3y ago
Nice! The README mentions `LayerNorm` is implemented here, but while it's in the equivalence tests with PyTorch, I don't see it in the implementation.
- dauertewigkeit 3y agoIt's part of the TensorLi definition where all the magic happens.
- joennlae 3y agoThat is true. I went for a simple implementation of the layer norm and included it in the tensorli definition. But it would have been better to define it as a moduli for clarity.