PyPI Stats

Search

All packages
Top packages

Track packages

pytest-jax-bench


PyPI page
Home page
Author: None
Summary: Pytest plugin to profile jitted JAX functions (compile time, runtime, memory).
Latest version: 0.5.0
Required dependencies: jax | numpy | nvtx | pytest
Optional dependencies: graphviz | matplotlib

Downloads last day: 2
Downloads last week: 17
Downloads last month: 54