Do you think it would be possible to implement the magic method __getitem__ for e.g. the circular buffer such that you can neatly index and slice as you would with any other array, instead of using the get_at_index method found here?
I'm not sure how these methods work in conjunction with jax transformations (jit, vmap, etc...)
Hi,
Do you think it would be possible to implement the magic method
__getitem__for e.g. the circular buffer such that you can neatly index and slice as you would with any other array, instead of using theget_at_indexmethod found here?I'm not sure how these methods work in conjunction with jax transformations (jit, vmap, etc...)