Merge pull request #532 from netket/PhilipVinc/jaximpro (f56fca6c) · 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

f56fca6c

Unverified Commit f56fca6c authored 4 years ago by Giuseppe Carleo Committed by GitHub 4 years ago

Downloads
- Plain Diff

Merge pull request #532 from netket/PhilipVinc/jaximpro

Jax improvements + remove outdtype

parents d3ab5e4d db3a1eb7

Loading

Changes 13

Hide whitespace changes

Inline Side-by-side

Showing

Test/Sampler/test_sampler.py 1 addition, 0 deletions

Test/Sampler/test_sampler.py
netket/machine/_jax_utils.py 97 additions, 27 deletions

netket/machine/_jax_utils.py
netket/machine/abstract_machine.py 1 addition, 12 deletions

netket/machine/abstract_machine.py
netket/machine/density_matrix/abstract_density_matrix.py 2 additions, 2 deletions

netket/machine/density_matrix/abstract_density_matrix.py
netket/machine/density_matrix/diagonal.py 0 additions, 1 deletion

netket/machine/density_matrix/diagonal.py
netket/machine/density_matrix/jax.py 4 additions, 6 deletions

netket/machine/density_matrix/jax.py
netket/machine/density_matrix/rbm.py 1 addition, 1 deletion

netket/machine/density_matrix/rbm.py
netket/machine/jax.py 10 additions, 32 deletions

netket/machine/jax.py
netket/machine/rbm.py 1 addition, 1 deletion

netket/machine/rbm.py
netket/machine/torch.py 2 additions, 2 deletions

netket/machine/torch.py
netket/operator/_der_local_values_jax.py 17 additions, 17 deletions

netket/operator/_der_local_values_jax.py
netket/operator/_local_cost_functions.py 6 additions, 12 deletions

netket/operator/_local_cost_functions.py
netket/vmc_common.py 94 additions, 70 deletions

netket/vmc_common.py

with 236 additions and 183 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