search

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX
Apache License 2.0
141 stars 9 forks source link

support fancier masks in levanter fa branch #15

Closed dlwh closed 11 months ago