nn: Infinity nn, jax and XLA
Infinity nn, jax and XLA
Google AI Blog: Fast and Easy Infinitely Wide Networks with Neural Tangents
↓ github
GitHub - google/neural-tangents: Fast and Easy Infinite Neural Networks in Python
Neural Tangents evaluate infinite networks as finite ones
(https://arxiv.org/abs/1912.02803)
↓ jax
simple nn functions using XLA(acceleration liner algebra), implimenting onto GPU and TPU etc.
↓ xla
Google Developers Blog: XLA - TensorFlow, compiled
XLA is a scaling a compil from space to time(AOT; ahead-of-time).
↓ github
tensorflow/tensorflow/compiler at master · tensorflow/tensorflow · GitHub