svae-jax pip install -e . Implementation of Composing graphical models with neural networks for structured representations and fast inference. Based on the original autograd repository.