Neural Tangents: Fast and Easy Infinite Neural Networks in Python