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
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
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
|