Skip to content

Commit 3813ba0

Browse files
committed
feat(wip): port reemannian hmc from research repo
Signed-off-by: Kai Xu <xuk@ibm.com>
1 parent 7fc04d2 commit 3813ba0

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

src/AdvancedHMC.jl

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,9 +142,17 @@ module Experimental
142142
using ..AdvancedHMC: @unpack, TYPEDEF, TYPEDFIELDS, AbstractScalarOrVec, AbstractLeapfrog, step, step_size
143143
import ..AdvancedHMC: ∂H∂θ, ∂H∂r, DualValue, PhasePoint, phasepoint, step
144144
include("riemannian/integrator.jl")
145-
include("riemannian/hamiltonian.jl")
146-
include("riemannian/metric.jl")
147145
export GeneralizedLeapfrog
146+
147+
import AdvancedHMC: _rand
148+
using AdvancedHMC: AbstractMetric, PhasePoint
149+
using LinearAlgebra: eigen, cholesky, Symmetric, Diagonal
150+
include("riemannian/metric.jl")
151+
export DenseRiemannianMetric
152+
153+
import AdvancedHMC: DualValue, phasepoint, neg_energy, ∂H∂θ, ∂H∂r
154+
using LinearAlgebra: logabsdet, tr
155+
include("riemannian/hamiltonian.jl")
148156
end
149157

150158
include("abstractmcmc.jl")

0 commit comments

Comments
 (0)