thatguy11325

Cache the number of elements in the action space

3

You probably dont need to dispatch to numpy everytime you call `split` to calculate the number of elements in the space. This PR caches the sizes (in a less than...

Should hopefully be faster. Based on my [comparison of different categorical distribution sampling methods](https://gist.github.com/thatguy11325/4df3b4d39e9b707b5ee0e09a7489769c). [Wandb tests](https://wandb.ai/thatguy11325/pufferlib/groups/puf-0.7.0-baseline/workspace?workspace=user-thatguy11325)