jax-flows
by ChrisWaites
Normalizing Flows in JAX 🌊
AI summary
Normalizing Flow Library
A Python library implementing normalizing flows, a type of generative model used in machine learning.
- stars
- 275
- forks
- 19
- watching
- 8
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Normalizing flow library
A library for building parametric models of complex distributions using normalizing flows and JAX.
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
Distribution modeler
An approach to modeling complex distributions by iteratively adding normalizing flow components and training with gradient boosting
Hydro model library
A Julia package for simulating hydrological processes in various configurations
Schema converter
Converts GraphQL schema to flow definition.
Numerical simulator framework
A JAX-based framework for creating differentiable numerical simulators with arbitrary discretizations.
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
Control Flow Image
A C compiler that translates code into an image-based control flow graph with implemented functionality
Data normalizer
Normalizes honeypot data from various feeds for analysis
JAX helper library
A set of utilities for writing reliable JAX code
Lie group library
A library that provides efficient and AD-friendly implementations of Lie groups for rigid body transformations in computer vision and robotics applications.
Traffic Optimizer
A framework for training reinforcement learning models to optimize traffic control and simulation
Graph library
A lightweight library for working with graph neural networks in jax.
Data graph library
A Python library for constructing dataflow graphs with support for reactive and lazy evaluation.
tfausak/flow200
Code readability enhancer
A Haskell package providing alternative operators to improve code readability
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jax-flows.
