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.0.tar.gz
(7.8 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.0-py3-none-any.whl
(10.7 kB
view details)
File details
Details for the file jaxfm-0.1.0.tar.gz.
File metadata
- Download URL: jaxfm-0.1.0.tar.gz
- Upload date:
- Size: 7.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.10.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
4d3663dcdb1ce6e1bc434f97dcaa8f63137706b33f0c109f945fbe4541ac9636
|
|
| MD5 |
e7ae6aaa466a2d7ef37e3fc8716e39c7
|
|
| BLAKE2b-256 |
e34fe5ae1ac249ce73d7b9db03a6ee5ee5d2549b6087af2ab44f9dbcd5f52c90
|
File details
Details for the file jaxfm-0.1.0-py3-none-any.whl.
File metadata
- Download URL: jaxfm-0.1.0-py3-none-any.whl
- Upload date:
- Size: 10.7 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 |
d21004757c43f034af60092288ff0a071d74a683bffc066e56f9ea0ee4a6bf4d
|
|
| MD5 |
bf367a7c18fa886c1ccb08dcd8e466bd
|
|
| BLAKE2b-256 |
3e2f13be756d47497a03cdce0700fb57a6fc9ea4a22eb9f80911ac4f3148cc57
|
