Lightwieght JAX context for managing collections of parameters and RNG sequences.
Project description
jaxctx
JAX Context for memoizationg collections of parameters, and handling sequences of random keys. Based loosely on things like haiku, and flax, but not deep learning specific. No support for lifting things like scan, etc. You must build these things on top of this if you want them.
Additional support added for probabilistic parameterisations based on priors.
Change Log
25 Feb, 2026 -- 1.1.0 released with scoped dicts structure changes. Breaks backward compatibility with 1.0.x, but adds support for nested contexts and more flexible scoping.
21 July, 2025 -- 1.0.3 released with support for jaxctx.prior and jaxctx.prior.Prior.
3 June, 2025 -- 1.0.2 prior constrained parameters released.
2 June, 2025 -- 1.0.1 released with context API.
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
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
File details
Details for the file jaxctx-1.1.1.tar.gz.
File metadata
- Download URL: jaxctx-1.1.1.tar.gz
- Upload date:
- Size: 32.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b854b64815fa1d40cbc00c898f3c11d1a13bc35de3f5500ec88d9e718ace620b
|
|
| MD5 |
78fabfa3bb00f0a3bd323df91dda5bd6
|
|
| BLAKE2b-256 |
c8723aef48d1deb45f6e87042fb752cd751c8944a81e273e199fea6d8d7d2302
|
File details
Details for the file jaxctx-1.1.1-py3-none-any.whl.
File metadata
- Download URL: jaxctx-1.1.1-py3-none-any.whl
- Upload date:
- Size: 28.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
7a6711ff309edcab604954b7065e8f66546c13965567aacd888b3e3c0bca394e
|
|
| MD5 |
11c3b355901d34d032eeca59fea6fcab
|
|
| BLAKE2b-256 |
ff2c80e6aee2a6bb3b3c493c18dd9751442a910d5c20ad193f88a9490b501267
|
