connex (pypi): Fine-grained, dynamic control of neural network topology in JAX.
Keywords: deep-learning|jax|neural-networks