search

stanford-crfm / haliax

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

Move StateDict to Haliax, clean it up a lot. #90

Open dlwh opened 1 month ago