Skip to content

[TE/JAX] XLA FFI calls for layer norm and RMS norm #4232

[TE/JAX] XLA FFI calls for layer norm and RMS norm

[TE/JAX] XLA FFI calls for layer norm and RMS norm #4232