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.3.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.3-py3-none-any.whl
(11.5 kB
view details)
File details
Details for the file jaxfm-0.1.3.tar.gz.
File metadata
- Download URL: jaxfm-0.1.3.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 |
e39d90c605c923459020d645870bae7cd459db1e574c8407a3d4c2795f5350b5
|
|
| MD5 |
107349d0fb2db40214d19d8e5d777a89
|
|
| BLAKE2b-256 |
48fbd9e7fae1d64384f2c5c64edeafcef16e19d86eb1c57a7fb7dc60f01d8312
|
File details
Details for the file jaxfm-0.1.3-py3-none-any.whl.
File metadata
- Download URL: jaxfm-0.1.3-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 |
d28d1bd100dbd0a1703599154aec58515448c40cef371d0cc366f77a1f653fd8
|
|
| MD5 |
1d1a40a5d58c014abab948b517806d14
|
|
| BLAKE2b-256 |
7d8896912a595fe22677dc78d31f1dd483847eb43f75554925e7c3110fff099c
|
