jax-autovmap

PyPI page
Home page
Author: None
Summary: Automatically broadcast inputs by dynamically applying jax.vmap given un-batched argument dimensions.
Latest version: 0.3.0
Required dependencies: jax | numpy

Downloads last day: 2
Downloads last week: 91
Downloads last month: 665