Simple Mixture Model This repo implement the code for Deep Learning with Jax and Elegy. Instalation With pip: pip install -r requirements.txt With poetry: poetry install Running the code python main.py