Google Neural Tangents - Fast & Easy Infinite Neural Networks in Py...