search

stanford-crfm / haliax

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

Add tree_map etc variants that set is_leaf=is_named_array by default #54

Closed dlwh closed 8 months ago

dlwh commented 8 months ago

fixed in dev