Laurent Mazare d32e8199cd Layer norm tweaks (#482) ...

* Add some options to make layer-norm more configurable.

* Add the rms-norm variant.

* Replace the RmsNorm with the shared bits.

2023-08-17 10:07:13 +01:00

..

main.rs

Add some tracing to the quantized example. (#473)

2023-08-16 18:49:08 +01:00

model.rs

Layer norm tweaks (#482)

2023-08-17 10:07:13 +01:00