Skip to main content

neural sampler

Project description

PyPI version Open In Colab

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 

Demonstrations and tutorials

Link Description
Open In Colab Train the neural sampler for the double well potential in PyTorch
Open In Colab Train the neural sampler for the Müller-Brown Model in PyTorch
Open In Colab Train the neural sampler for the periodic potential in PyTorch
Open In Colab Train the neural sampler for the Ginzburg-Landau potential in PyTorch

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
  • plz check the abf-mmd can reproduce the results! I have checked the codes are the same? I guessed the only issue might be just run enough runs! (Sun 09/05/2021 21:06)
  • at least lots of things are connected now!
  • update the colab module, going to download the dataset from the cloud!

Project details


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.14.tar.gz (22.0 kB view hashes)

Uploaded Source

Built Distribution

neuralsampler-0.0.14-py3-none-any.whl (25.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page