Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Operator interface #76

Merged
merged 5 commits into from
Sep 13, 2024
Merged

Operator interface #76

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
5e21633
Implemented required operators.
Sep 12, 2024
0e555f0
Added test.
Sep 12, 2024
612d37b
Some cleanup.
Sep 12, 2024
d07e231
Fixed #72
Sep 12, 2024
6c35f19
Apply operators in natural order. Some cleanup.
Sep 13, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions examples/ex0_ground_state_search.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,7 @@
"sys.path.append(sys.path[0] + \"/..\")\n",
"\n",
"import jax\n",
"from jax.config import config\n",
"config.update(\"jax_enable_x64\", True)\n",
"jax.config.update(\"jax_enable_x64\", True)\n",
"\n",
"import jax.random as random\n",
"import jax.numpy as jnp\n",
Expand Down
3 changes: 1 addition & 2 deletions examples/ex0_ground_state_search.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,7 @@
# coding: utf-8

import jax
from jax.config import config
config.update("jax_enable_x64", True)
jax.config.update("jax_enable_x64", True)

import jax.random as random
import jax.numpy as jnp
Expand Down
7 changes: 1 addition & 6 deletions examples/ex2_unitary_time_evolution.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,12 +1,7 @@
import os

import jax
from jax.config import config
config.update("jax_enable_x64", True)

import jax.random as random
import flax
import jax.numpy as jnp
jax.config.update("jax_enable_x64", True)

import numpy as np

Expand Down
1 change: 0 additions & 1 deletion examples/ex3_custom_net.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import jax
import flax
import jVMC
import numpy as np

# This class defines the network structure of a complex RBM

Expand Down
3 changes: 1 addition & 2 deletions examples/ex4_benchmarking.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,7 @@
import os

import jax
from jax.config import config
config.update("jax_enable_x64", True)
jax.config.update("jax_enable_x64", True)

import jax.random as random
import jax.numpy as jnp
Expand Down
6 changes: 1 addition & 5 deletions examples/ex5_dissipative_Lindblad.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,13 +1,9 @@
import matplotlib.pyplot as plt
import numpy as np
import flax
import jax.random as random
from jax.config import config
import jax.numpy as jnp
import jax
import jVMC
from functools import partial
config.update("jax_enable_x64", True)
jax.config.update("jax_enable_x64", True)


def copy_dict(a):
Expand Down
6 changes: 1 addition & 5 deletions examples/ex6_dissipative_Lindblad_2D.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,12 +1,8 @@
import matplotlib.pyplot as plt
import numpy as np
import flax
import jax.random as random
from jax.config import config
import jax.numpy as jnp
import jax
import jVMC
config.update("jax_enable_x64", True)
jax.config.update("jax_enable_x64", True)
from functools import partial


Expand Down
11 changes: 5 additions & 6 deletions examples/ex7_fermions.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,7 @@
"\n",
"# jax\n",
"import jax\n",
"from jax.config import config\n",
"config.update(\"jax_enable_x64\", True)\n",
"jax.config.update(\"jax_enable_x64\", True)\n",
"import jax.numpy as jnp\n",
"import flax.linen as nn\n",
"\n",
Expand Down Expand Up @@ -138,11 +137,11 @@
" if i == flavourL//flavour-1:\n",
" continue\n",
" # up chain hopping\n",
" hamiltonian.add(op.scal_opstr( t, ( annihilation(site1UP + i) ,creation(site1UP + i + 1) ) ) )\n",
" hamiltonian.add(op.scal_opstr( t, ( annihilation(site1UP + i + 1) ,creation(site1UP + i) ) ) )\n",
" hamiltonian.add(op.scal_opstr( t, ( creation(site1UP + i + 1), annihilation(site1UP + i) , ) ) )\n",
" hamiltonian.add(op.scal_opstr( t, ( creation(site1UP + i), annihilation(site1UP + i + 1) , ) ) )\n",
" # down chain hopping\n",
" hamiltonian.add(op.scal_opstr( t, ( annihilation(site1DO - i) ,creation(site1DO - i - 1) ) ) )\n",
" hamiltonian.add(op.scal_opstr( t, ( annihilation(site1DO - i - 1) ,creation(site1DO - i) ) ) )"
" hamiltonian.add(op.scal_opstr( t, ( creation(site1DO - i - 1), annihilation(site1DO - i) , ) ) )\n",
" hamiltonian.add(op.scal_opstr( t, ( creation(site1DO - i), annihilation(site1DO - i - 1) , ) ) )"
]
},
{
Expand Down
Loading
Loading