Matteo Hessel
Matteo Hessel
This is really great!
Thank you very much! If the paper claims stack up, this will be very useful to the jax community. Btw there is pytorch reference code from the authors of the...
Apologies for the long delay, would you mind moving this to `contrib/`? Also would you consider implementing the wrapper version instead?
Do you have a reference to this specific way of normalising?
Sounds like it could be a good addition. Do you want to put together a PR?
Sounds like a nice contribution, do you want to take a stab at it?
Hello. I was on paternity leave for most of the past year. Are you still having this issue? Happy to look into it if that's the case
which arguments are causing errors to you?
Huge apologies for the long delay here, if you can sync we get this submitted?
@Sea-Snell, as @mkunesch mentioned it would be helpful to have a minimal example we could try this with