neural sampler
Project description
NeuralSampler - Pytorch
Implementation of Neural Sampler.
Install
$ pip install neuralsampler
or install the latest version by
pip install -U git+https://JiahaoYao:{password}@github.com/JiahaoYao/neuralsampler.git@main
Install the jax (follow the official instruction here)
pip install jax jaxlib==0.1.64+[YOUR_CUDA_VERSION] -f https://storage.googleapis.com/jax-releases/jax_releases.html
e.g.
pip install --upgrade jax jaxlib==0.1.64+cuda101 -f https://storage.googleapis.com/jax-releases/jax_releases.html
Usage
import torch
from neuralsampler import neuralsampler
Run the code
python main.py
Run the scripts
python scripts/test_job.py
Ignore me (random things)
this repo is to collect all the random results and reproduce the experiments here
and then jax
i will use jax and flax, like shown here: https://github.com/yang-song/score_sde
there are the templates of building the jax neural networks (quite interesting to try this functional programming )
Todo list
- this library is on the pytorch
- i am also going to prepare the colab notebook
- the dataset is through the gdown: you can download the dataset from the google drive
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
neuralsampler-0.0.9.tar.gz
(17.0 kB
view hashes)
Built Distribution
Close
Hashes for neuralsampler-0.0.9-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5e9836620eb022f2aa738303ff44fa2f8acf3d90693f6dbec22fd2f38ef95a58 |
|
MD5 | 267ea48984b382be00a6222c156cb5df |
|
BLAKE2b-256 | f670ea2a47b1c69f7940904ac2eb72f186b818dcd5fc98d2ca9d86e35f91598a |