Test and make an example using AEStream to load data into Spyx. An interesting question is whether AEStream can write to JAX tensors being used by Spyx in a compiled loop? https://github.com/aestream/aestream/tree/v0.6.2