zlax

Python and Zelus interfaces for the Zelus to Python-JAX compiler and runtime libraries. zlax supports also ProbZelus and provides interfaces and runtime librairies to write probabilistic models and run inference.

# zlax and probzlax
pip install ./zlax
pip install ./probzlax

# zelus-jax
opam pin -k path .

# Zelus
zeluc -I `zeluc -where`-jax -jax <name>.zls

# ProbZelus
probzeluc -I `zeluc -where`-jax -jax <name>.zls

Run zluciole. See usage using the -h option.