make logpdf a pytree in sampler (866ee1ce) · Commits · Vicentini Filippo / netket · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

N

netket
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

866ee1ce

Commit 866ee1ce authored 1 year ago by Filippo Vicentini

Downloads
- Patches
- Plain Diff

make logpdf a pytree in sampler

cleanup


impro


black

parent b970084d

Branches pv/sinisample

No related tags found

1 merge request!1434Support PyTree - logpdf in sampler

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

netket/sampler/exact.py 1 addition, 1 deletion

netket/sampler/exact.py
netket/sampler/metropolis.py 1 addition, 1 deletion

netket/sampler/metropolis.py
netket/sampler/metropolis_numpy.py 1 addition, 2 deletions

netket/sampler/metropolis_numpy.py
netket/utils/jax.py 38 additions, 4 deletions

netket/utils/jax.py
netket/utils/partial.py 8 additions, 6 deletions

netket/utils/partial.py

with 49 additions and 14 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment