vmap
A package designed for fast training and inference for large numbers of small neural networks simultaneously.