{"library":"drjax","type":"library","category":null,"description":"DrJAX is a Python library built on JAX that provides scalable and differentiable MapReduce primitives. It enables users to express complex computations over distributed data in a functional, JAX-compatible manner, allowing for automatic differentiation and XLA compilation across various hardware accelerators. The current version is 0.1.4, and it sees regular, minor updates focusing on stability and JAX compatibility.","language":"python","status":"active","version":"0.1.4","tags":["jax","mapreduce","distributed-computing","machine-learning","differentiation","google"],"last_verified":"Tue May 26","install":[{"cmd":"pip install drjax","imports":["from drjax import map_reduce","from drjax import map","from drjax import broadcast"]}],"homepage":null,"github":"https://github.com/google/drjax","docs":null,"changelog":null,"pypi":"https://pypi.org/project/drjax/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":50,"avg_install_s":13,"avg_import_s":null,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/drjax/compatibility"}}