jax-flows
Normalizing Flow Library
A Python library implementing normalizing flows, a type of generative model used in machine learning.
Normalizing Flows in JAX 🌊
274 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 | 461 |
 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 | 120 |
 joarwilk/gql2flow | Converts GraphQL schema to flow definition. | 97 |
 ucl-bug/jaxdf | A JAX-based framework for creating differentiable numerical simulators with arbitrary discretizations. | 121 |
 google-deepmind/tf2jax | Converts TensorFlow functions to equivalent JAX Python functions. | 105 |
 juliapoo/artfuscator | A C compiler that translates code into an image-based control flow graph with implemented functionality | 1,011 |
 johnnykv/mnemosyne | Normalizes honeypot data from various feeds for analysis | 45 |
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 788 |
 brentyi/jaxlie | A library that provides high-level implementations of Lie groups commonly used for rigid body transformations in computer vision and robotics applications. | 233 |
 flow-project/flow | A framework for training reinforcement learning models to optimize traffic control and simulation | 1,068 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,375 |
 streamlet-dev/tributary | A Python library for constructing dataflow graphs with support for reactive and lazy evaluation. | 442 |
 tfausak/flow | A package providing functions and operators to improve Haskell code readability | 200 |