search

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX
Apache License 2.0
149 stars 11 forks source link

fix hax.where in the presence of scalar NamedArrays #86

Closed dlwh closed 5 months ago