/haliax

Named Tensors for Legible Deep Learning in JAX

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers