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:
40
Downloads last week:
232
Downloads last month:
1,171