narrow range where we use jit for sharding - Githubissues

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX

Apache License 2.0

146 stars 11 forks source link

narrow range where we use jit for sharding #98

Closed dlwh closed 1 month ago