Merge pull request #554 from netket/numba4jax (230725c2) · 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

230725c2

Unverified Commit 230725c2 authored 4 years ago by Giuseppe Carleo Committed by GitHub 4 years ago

Downloads
- Plain Diff

Merge pull request #554 from netket/numba4jax

Numba4jax: Numba in HamiltonianSampler

parents 99c73a4d ae92739b

Loading

Changes 13

Hide whitespace changes

Inline Side-by-side

Showing

Examples/Ising1d/ising1d.py 1 addition, 1 deletion

Examples/Ising1d/ising1d.py
Test/Sampler/test_sampler.py 6 additions, 0 deletions

Test/Sampler/test_sampler.py
netket/jax/__init__.py 2 additions, 0 deletions

netket/jax/__init__.py
netket/jax/numba4jax.py 343 additions, 0 deletions

netket/jax/numba4jax.py
netket/operator/_abstract_operator.py 11 additions, 0 deletions

netket/operator/_abstract_operator.py
netket/operator/_hamiltonian.py 40 additions, 0 deletions

netket/operator/_hamiltonian.py
netket/operator/_local_operator.py 29 additions, 0 deletions

netket/operator/_local_operator.py
netket/operator/_pauli_strings.py 33 additions, 2 deletions

netket/operator/_pauli_strings.py
netket/sampler/__init__.py 4 additions, 2 deletions

netket/sampler/__init__.py
netket/sampler/exact.py 3 additions, 3 deletions

netket/sampler/exact.py
netket/sampler/metropolis.py 3 additions, 0 deletions

netket/sampler/metropolis.py
netket/sampler/rules/hamiltonian.py 39 additions, 25 deletions

netket/sampler/rules/hamiltonian.py
netket/sampler/rules/hamiltonian_numpy.py 6 additions, 2 deletions

netket/sampler/rules/hamiltonian_numpy.py

with 520 additions and 35 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