AdaCat
pip package for adaptive categorical distribution (Adacat) proposed in ADACAT: Adaptive Categorical Discretization for Autoregressive Models.
- installation:
pip install adacat
- see usage in
demo-(jax/torch).py
/demo-(jax/torch).ipynb
Citation
The bibtex is provided below for citation covenience.
@inproceedings{
li2022adacat,
title={AdaCat: Adaptive Categorical Discretization for Autoregressive Models},
author={Qiyang Li and Ajay Jain and Pieter Abbeel},
booktitle={The 38th Conference on Uncertainty in Artificial Intelligence},
year={2022},
url={https://openreview.net/forum?id=HMzzPOLs9l5}
}
Acknowledgements
Thank Ilya Kostrikov for help on the JAX implementation of Adacat.