Neural Tangents - Linearizing Neural Network training.

Neural Tangents is a set of tools that can be used to probe the linearized training dynamics of neural networks. There are two, dual, perspectives that are explored here: linearization of training in weight space, and linearization of training in function space.

For more details, visit the source.