search

stanford-crfm / haliax

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

add a non-scan-layers version of Stacked #70

Closed dlwh closed 3 months ago

dlwh commented 4 months ago

seems like we're moving away from scan layers?