PyPI page
Home page
Author:
None
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
...
Summary:
DrJAX - Scalable and Differentiable MapReduce Primitives in JAX.
Latest version:
0.1.4
Required dependencies:
absl-py
|
jax
|
jaxlib
Downloads last day:
7,960
Downloads last week:
58,973
Downloads last month:
176,283