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.1.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.1-py3-none-any.whl
(10.7 kB
view details)
File details
Details for the file jaxfm-0.1.1.tar.gz.
File metadata
- Download URL: jaxfm-0.1.1.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 |
d25067aa6fe4f73298d3909c1aa2a3cfd977d448ff85c205e6f7c822076eba71
|
|
| MD5 |
c1d4a200fb88238abf5550241de08c8c
|
|
| BLAKE2b-256 |
6bf44a183c3db1dc19cb374e906f351d161f78baeedcfe51c9159acc21056d5f
|
File details
Details for the file jaxfm-0.1.1-py3-none-any.whl.
File metadata
- Download URL: jaxfm-0.1.1-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 |
4c72e30a1b1f3e81e0884c1f0d904c8e165df2651314a366f8951cef2ea0a1ae
|
|
| MD5 |
075f3d5e95714f4ccc342466f5c5394c
|
|
| BLAKE2b-256 |
f9a0609051a598d5ffe843aa5d8c6340699b87b749a3336444eba03e7a75f46e
|
