JAX Talk: Generating Extremely Long Sequences with S4

In this session, our guest speaker Sasha Rush will be teaching us how to create extremely long sequences with S4. This is a part of the JAX Talks and we will get to see some usage of JAX + Flax