Skip to content

Commit

Permalink
feat(wip): port reemannian hmc from research repo
Browse files Browse the repository at this point in the history
Signed-off-by: Kai Xu <[email protected]>
  • Loading branch information
xukai92 committed Jul 25, 2024
1 parent 7fc04d2 commit 3813ba0
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions src/AdvancedHMC.jl
Original file line number Diff line number Diff line change
Expand Up @@ -142,9 +142,17 @@ module Experimental
using ..AdvancedHMC: @unpack, TYPEDEF, TYPEDFIELDS, AbstractScalarOrVec, AbstractLeapfrog, step, step_size
import ..AdvancedHMC: ∂H∂θ, ∂H∂r, DualValue, PhasePoint, phasepoint, step
include("riemannian/integrator.jl")
include("riemannian/hamiltonian.jl")
include("riemannian/metric.jl")
export GeneralizedLeapfrog

import AdvancedHMC: _rand
using AdvancedHMC: AbstractMetric, PhasePoint
using LinearAlgebra: eigen, cholesky, Symmetric, Diagonal
include("riemannian/metric.jl")
export DenseRiemannianMetric

import AdvancedHMC: DualValue, phasepoint, neg_energy, ∂H∂θ, ∂H∂r
using LinearAlgebra: logabsdet, tr
include("riemannian/hamiltonian.jl")
end

include("abstractmcmc.jl")
Expand Down

0 comments on commit 3813ba0

Please sign in to comment.