search

stanford-crfm / haliax

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

NN Layers: RNN Backbone #26

Open dlwh opened 11 months ago

dlwh commented 11 months ago

I think this just wraps an RNN Cell module and calls scan on it.