JAX-based implementation of Flow Matching
Project description
JAXFM : A Python package for Flow Matching using JAX
Description
A Python package implementing flow matching methods using jax. The code includes
reflows.py: Implementation of [1]conditional.py: Implementation of [2] Various test datasets are also available to test the above methods.
Refrences
[1]: (Liu et al.) Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow
[2]: (Lipman et al.) Flow Matching for Generative Modeling
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
jaxfm-0.1.2a0.tar.gz
(8.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
jaxfm-0.1.2a0-py3-none-any.whl
(11.5 kB
view details)
File details
Details for the file jaxfm-0.1.2a0.tar.gz.
File metadata
- Download URL: jaxfm-0.1.2a0.tar.gz
- Upload date:
- Size: 8.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.10.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8635e51ff6e9f5c602f036c77dec340834edbdbdd690649fcce5cf16664ae35e
|
|
| MD5 |
579e9137931e114f18fa387f692fadfd
|
|
| BLAKE2b-256 |
2035bd385d30bdf3788834f243c420909074826ace1f5272ba1b8e2c3563af98
|
File details
Details for the file jaxfm-0.1.2a0-py3-none-any.whl.
File metadata
- Download URL: jaxfm-0.1.2a0-py3-none-any.whl
- Upload date:
- Size: 11.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.10.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
750d74e5a57feddb5154967aef0fc6a983b75730b989e829b1bb80196a08a14e
|
|
| MD5 |
7527756c07441ffbde20bacac68ae14c
|
|
| BLAKE2b-256 |
6885786d245a8ce767a9f7b4da6f5254641aa2538b62011cfcc6f40ea8b6de71
|
