Skip to content

Small bugfix in langevin sampler

Vicentini Filippo requested to merge github/fork/jwnys/langevinfix into master

Created by: jwnys

Bug fixes to Langevin sampler:

  • when providing chunk_size != None, we got a tracing error
  • for some models the grad function failed due to a shape (1,)

Merge request reports