8 citations · 19 across the 9 of their papers we have counts for
Showing 2024Show all
2 papers · 1 filter
cs.LG2024
Batch size invariant Adam
Xi Wang, Laurence Aitchison
We propose a batch size invariant version of Adam, for use in large-scale, distributed settings, in which the mini-batch is divided into micro-batches which are distributed among w…
cs.MS2024★ 8 cited
BlackJAX: Composable Bayesian inference in JAX
Alberto Cabezas, Adrien Corenflos, Junpeng Lao +19
BlackJAX is a library implementing sampling and variational inference algorithms commonly used in Bayesian computation. It is designed for ease of use, speed, and modularity by tak…