tree 932b37191ae544492ba712376c9f33555694d100
parent 7227a8b7bee47beda92f7ee084f9abec40a59d6c
author Austin Schuh <austin.linux@gmail.com> 1726962838 -0700
committer Austin Schuh <austin.linux@gmail.com> 1726964132 -0700
gpgsig -----BEGIN PGP SIGNATURE-----
 
 iQGzBAABCgAdFiEExcNA7Sy+j6Ly5hA4QVYw08sOnWYFAmbvYaQACgkQQVYw08sO
 nWbAPAwAm1s/5M8C3pu+7MtCa28IKjYj9+l1rhSUgLqKrfq4MGkCbaUJF71HukVj
 eCkhlfCp6L4zs3xL/u55gT/I3KUE8499c94dO5XFy+byAB7RAFCFCuFrjWxC3n8v
 bEjbfr5rohGbCGDPPBEGz/bVP3znRd4Khd81av2OuqwN91tot4UMISfTnqazU2ik
 rwT7ho6HzHKu7rQLdkPqlXmT1KFIQbHPLLvLnYb56VjESlHW4FjWM67puNUIai6H
 R8VVQsv3IA18+eVKxw90k8B9/iyE5iewcxBhRSwQpQQOWXaHdwsMihLAw5f3HCF+
 +QA00gGEFhoGypEtA6fJIkpaQCEVaY5Ti6aeIANKPdSDCYT8HvOxq6kdKaMIh9sJ
 X+OEc6f+M+qRtRIILx9Nkhk66mQsD2FIzsYqoOuUJIa1kdderJUIbQx1G8AqpsUj
 lpFSdzpALneV9FtQE9BInF3l4Nr0MNdB0wqi004/ia1dnJtbE59Uejh8QWNkrb30
 vCuvvRP4
 =soiP
 -----END PGP SIGNATURE-----

Add flashbax support

This gives us a JAX experience buffer for Soft Actor-Critic.

Add jaxtyping in too so we can type things better.

Change-Id: I4c9f0071f9e912dcab0c883da8d9d7990ed06c46
Signed-off-by: Austin Schuh <austin.linux@gmail.com>