search

stanford-crfm / haliax

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

better support for the `inference` field in dropout #30

Closed dlwh closed 1 year ago

dlwh commented 1 year ago

a bunch of dumb small limitations in the way I had set things up