Implementation of Score-Based Deterministic Density Sampling

Folder sbtm contains the implementations of SBTM, SDE and SVGD samplers. File scripts/8-vi-generate-data.py generates the data. File scripts/9-vi-visualize-data.py does the visualization.

Dependencies: Jax, Optax, Flax, seaborn, matplotlib, tqdm.