neural-tangents (pypi): Fast and Easy Infinite Neural Networks in Python
Keywords: bayesian-inference|bayesian-networks|deep-networks|gaussian-processes|gradient-descent|gradient-flow|infinite-networks|jax|kernel|kernel-computation|neural-networks|neural-tangents|training-dynamics