Expose the filter_checkpoint helper: it's useful for custom training loops. - Githubissues

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX

Apache License 2.0

140 stars 9 forks source link

Expose the filter_checkpoint helper: it's useful for custom training loops. #91

Closed rjpower closed 1 month ago