jax-flows
Normalizing Flow Library
A Python library implementing normalizing flows, a type of generative model used in machine learning.
Normalizing Flows in JAX 🌊
275 stars
8 watching
19 forks
Language: Python
last commit: over 1 year ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 information-fusion-lab-umass/nux | A library for building parametric models of complex distributions using normalizing flows and JAX. | 82 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 467 |
 robert-giaquinto/gradient-boosted-normalizing-flows | An approach to modeling complex distributions by iteratively adding normalizing flow components and training with gradient boosting | 27 |
 deltares/wflow.jl | A Julia package for simulating hydrological processes in various configurations | 122 |
 joarwilk/gql2flow | Converts GraphQL schema to flow definition. | 97 |
 ucl-bug/jaxdf | A JAX-based framework for creating differentiable numerical simulators with arbitrary discretizations. | 124 |
 google-deepmind/tf2jax | Converts TensorFlow functions to equivalent JAX Python functions. | 109 |
 juliapoo/artfuscator | A C compiler that translates code into an image-based control flow graph with implemented functionality | 1,033 |
 johnnykv/mnemosyne | Normalizes honeypot data from various feeds for analysis | 46 |
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 797 |
 brentyi/jaxlie | A library that provides efficient and AD-friendly implementations of Lie groups for rigid body transformations in computer vision and robotics applications. | 236 |
 flow-project/flow | A framework for training reinforcement learning models to optimize traffic control and simulation | 1,078 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |
 streamlet-dev/tributary | A Python library for constructing dataflow graphs with support for reactive and lazy evaluation. | 444 |
 tfausak/flow | A Haskell package providing alternative operators to improve code readability | 200 |