PyPI page
Home page
Author:
None
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTI...
Summary:
DrJAX - Scalable and Differentiable MapReduce Primitives in JAX.
Latest version:
0.2.0
Required dependencies:
absl-py
|
jax
|
jaxlib
Downloads last day:
7,546
Downloads last week:
55,268
Downloads last month:
261,994