enkeejunior1/TP4

Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)

PythonApache-2.0