update project.toml

Author: GiggleLiu

Project.toml

@@ -6,6 +6,8 @@ version = "0.2.0"
 [deps]
 Compose = "a81c6b42-2e10-5240-aca2-a61377ecd94b"
 DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab"
+HDF5 = "f67ccb44-e63f-5c2f-98bd-6dc0ccc4ba2f"
+LightGraphs = "093fc24a-ae57-5d10-9952-331d41423f4d"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LuxurySparse = "d05aeea4-b7d4-55ac-b691-9e7fabb07ba2"
 NiLang = "ab4ef3a6-0b42-11ea-31f6-e34652774712"
@@ -22,6 +24,8 @@ Yao = "5872b779-8223-5990-8dd0-5abbb0748c8c"
 Compose = "0.9"
 LuxurySparse = "0.6"
 NiLang = "0.8"
+LightGraphs = "1.3"
+HDF5 = "0.15"
 Requires = "1"
 SimpleTensorNetworks = "0.1"
 TropicalNumbers = "0.2.1"

src/pluto.jl

@@ -1,6 +1,6 @@
 using .Pluto
 
-@info "You can use the notebooks now by typing, e.g.
+@info "You can use the notebooks in `TropicalTensors` now by typing, e.g.
   ∘ `TropicalTensors.notebook(\"spinglass\")`, solving square lattice spinglass using a quantum simulator.
   ∘ `TropicalTensors.notebook(\"ising_and_2sat\")`, solving Ising spinglass and 2-SAT counting using tensor network contraction.
 "

test/second_neighbor.jl

@@ -61,3 +61,21 @@ end
     sg = Spinglass(lt, [-1,1], [0,0,0])
     @test solve(sg).n == 2
 end
+
+@testset "forwarddiff" begin
+    L = 6
+    lt = rand_maskedsquare(L, L, 0.8)
+    hs = zeros(length(lt))
+    Js = rand([-1,1.0], length(sgbonds(lt)))
+    res = solve(lt, Js, hs; usecuda=false).n
+    gs = ForwardDiff.gradient(x->(dn = solve(lt, eltype(x).(Js), x; usecuda=false).n; (@test ForwardDiff.value(dn)==res); dn), hs)
+    function compute_loss(lt, Js, config)
+        res = 0.0
+        vs = vertices(lt)
+        for (i, (src, dst)) in enumerate(sgbonds(lt))
+            res += Js[i] * config[findfirst(==(src), vs)] * config[findfirst(==(dst), vs)]
+        end
+        return res
+    end
+    @test compute_loss(lt, Js, gs) == res
+end