Neural Tangents: Fast and Easy Infinite Neural Networks in Python — mentioned by alevskaya

Fri, 31 Jan 2020 07:28:57 GMT

It's still early days for JAX, but there's neural tangents https://arxiv.org/abs/1912.02803 and reformer https://arxiv.org/abs/2001.04451 from iclr.

Paper and discussion history. A mention is not an endorsement.

Neural Tangents: Fast and Easy Infinite Neural Networks in Python · HackerPapers discussions