Module batch 

Lattice sums over arrays of inputs, with their gradients.

treams-rs extension: the lsum* ufuncs of treams broadcast their inputs but have no gradients.

Structs

SumResidual

What sum_array saves for a batch of lattice sums: the original inputs. Every input holds one shared value or one value per output.

Constants

PARALLEL 🔒

Batched sums split adaptively across the rayon pool from this many outputs.

Functions

sum_array

sum_part over arrays, with the residual for the pullback.