XMaster96 issues

Repositories
Issues
Comments

Results 2 issues of


                                            XMaster96

improve inputs.py

- add datasets.window to inputs.py

Categorical distributions / JAX vmap Error

I am trying to use a Categorical distributions inside of a function with JAX vmap. I also tested pmap and jit and both are working fine. Thanks for the help....