Haliax: Named Tensors for Legible Deep Learning in Jaxgithub.com/stanford-crfm2 pointscl3misch3 years ago