Skip to main content

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


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)

Uploaded Source

Built Distribution

multigrad-1.0.0-py3-none-any.whl (15.8 kB view details)

Uploaded Python 3

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

Hashes for multigrad-1.0.0.tar.gz
Algorithm Hash digest
SHA256 4cd4d88b66a3565461f20acc4853c7414a6f747130bb1b35020991091a79a41e
MD5 09b75a78c36f21a081b7523197606af1
BLAKE2b-256 1b4c07680d26c04bc31842d9f0e0e716ca03e8bceaa8f720ebd821698e6c6951

See more details on using hashes here.

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

Hashes for multigrad-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 816b2bd6d5a18dc96105b1b5217f659e44d79588f7e38bc4d46a3545dfb372cb
MD5 29a5e6ef7cbf4f41f1c361773eb6265e
BLAKE2b-256 96053e4e4036fe01a6b63fa167d20ebcfd2a96814aaab702ceddadaf8d83466d

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page