kdescent Stochastic gradient descent via mini-batched KDE Author Alan Pearl Prerequisites JAX (conda install -c conda-forge jax) Installation pip install --upgrade git+https://github.com/AlanPearl/kdescent.git