Differentiable Multiprocessing for Gradient Descent with JAX
Project description
multigrad
Framework to implement JAX models that can be distributed over MPI
Author
- Alan Pearl
Documentation
Online documentation is available at multigrad.readthedocs.io.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
multigrad-1.0.0.tar.gz
(237.4 kB
view details)
Built Distribution
multigrad-1.0.0-py3-none-any.whl
(15.8 kB
view details)
File details
Details for the file multigrad-1.0.0.tar.gz
.
File metadata
- Download URL: multigrad-1.0.0.tar.gz
- Upload date:
- Size: 237.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4cd4d88b66a3565461f20acc4853c7414a6f747130bb1b35020991091a79a41e |
|
MD5 | 09b75a78c36f21a081b7523197606af1 |
|
BLAKE2b-256 | 1b4c07680d26c04bc31842d9f0e0e716ca03e8bceaa8f720ebd821698e6c6951 |
File details
Details for the file multigrad-1.0.0-py3-none-any.whl
.
File metadata
- Download URL: multigrad-1.0.0-py3-none-any.whl
- Upload date:
- Size: 15.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 816b2bd6d5a18dc96105b1b5217f659e44d79588f7e38bc4d46a3545dfb372cb |
|
MD5 | 29a5e6ef7cbf4f41f1c361773eb6265e |
|
BLAKE2b-256 | 96053e4e4036fe01a6b63fa167d20ebcfd2a96814aaab702ceddadaf8d83466d |