PyPI Stats

Search

All packages
Top packages

Track packages

jax-autovmap


PyPI page
Author: Mathis Gerdes
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: 5
Downloads last week: 90
Downloads last month: 410
7-day smoothing