GitHub repository: mmhamdy/neural-tangents
Description: Fast and Easy Infinite Neural Networks in Python