search

stanford-crfm / haliax

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

No default in_resources #94

Closed dlwh closed 3 weeks ago

dlwh commented 3 weeks ago

@rjpower looks like the issue is fixed so this is a cleaner solution to the mesh issue