search

stanford-crfm / haliax

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

Mixed Precision and Resource Envs #65

Open dlwh opened 9 months ago