Use `inference` field a la equinox - Githubissues

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX

Apache License 2.0

141 stars 9 forks source link

Use `inference` field a la equinox #29

Closed dlwh closed 10 months ago