-
Notifications
You must be signed in to change notification settings - Fork 124
/
packages.yml
154 lines (139 loc) · 6.13 KB
/
packages.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
- title: FrEIA
date: 2018-09-07
url: https://github.com/VLL-HD/FrEIA
authors:
- name: VLL Heidelberg
url: https://hci.iwr.uni-heidelberg.de/vislearn
lang: PyTorch
description: The Framework for Easily Invertible Architectures (FrEIA) is based on RNVP flows. Easy to setup, it allows to define complex Invertible Neural Networks (INNs) from simple invertible building blocks.
- title: nflows
date: 2020-02-09
url: https://github.com/bayesiains/nflows
authors:
- name: Bayesiains
url: https://homepages.inf.ed.ac.uk/imurray2/group
lang: PyTorch
description: A suite of most of the SOTA methods using PyTorch. From an ML group in Edinburgh. They created the current SOTA spline flows. Almost as complete as you'll find from a single repo.
- title: flowtorch
date: 2020-12-07
url: https://github.com/facebookincubator/flowtorch
authors:
- name: Facebook / Meta
url: https://opensource.fb.com
lang: PyTorch
description: FlowTorch is a PyTorch library for learning and sampling from complex probability distributions using Normalizing Flows.
- title: TensorFlow Probability
date: 2018-06-22
url: https://github.com/tensorflow/probability
authors:
- name: Google
url: https://tensorflow.org/probability
lang: TensorFlow
description: Large first-party library that offers RNVP, MAF among other autoregressive models plus a collection of composable bijectors.
- title: NuX
date: 2020-03-09
url: https://github.com/Information-Fusion-Lab-Umass/NuX
authors:
- name: Information Fusion Labs (UMass)
lang: JAX
description: A library that offers normalizing flows using JAX as the backend. Has some SOTA methods. They also feature a surjective flow via quantization.
- title: jax-flows
date: 2020-03-23
url: https://github.com/ChrisWaites/jax-flows
authors:
- name: Chris Waites
url: https://chriswaites.com
lang: JAX
description: Another library that has normalizing flows using JAX as the backend. Has some of the SOTA methods.
- title: Distrax
date: 2021-04-12
url: https://github.com/deepmind/distrax
authors:
- name: DeepMind
url: https://deepmind.com
github: https://github.com/google-deepmind
lang: JAX
description: Distrax is a lightweight library of probability distributions and bijectors. It acts as a JAX-native re-implementation of a subset of TensorFlow Probability (TFP), with some new features and emphasis on extensibility.
- title: pzflow
date: 2021-06-17
url: https://github.com/jfcrenshaw/pzflow
authors:
- name: John Franklin Crenshaw
url: https://jfcrenshaw.github.io
lang: JAX
description: A package that focuses on probabilistic modeling of tabular data, with a focus on sampling and posterior calculation.
- title: InvertibleNetworks.jl
date: 2020-02-07
url: https://github.com/slimgroup/InvertibleNetworks.jl
authors:
- name: SLIM
url: https://slim.gatech.edu
lang: Julia
description: A Flux compatible library implementing invertible neural networks and normalizing flows using memory-efficient backpropagation. Uses manually implemented gradients to take advantage of the invertibility of building blocks, which allows for scaling to large-scale problem sizes.
- title: Zuko
date: 2022-05-21
date_added: 2022-10-19
last_updated: 2022-10-19
url: https://github.com/francois-rozet/zuko
authors:
- name: François Rozet
url: https://francois-rozet.github.io
lang: PyTorch
description: |
Zuko is a Python package that implements normalizing flows in PyTorch. It relies heavily on PyTorch's built-in distributions and transformations, which makes the implementation concise, easy to understand and extend. The API is fully documented with references to the original papers.
Zuko is used in [LAMPE](https://github.com/francois-rozet/lampe) to enable Likelihood-free AMortized Posterior Estimation with PyTorch.
- title: Jammy Flows
date: 2021-01-25
url: https://github.com/thoglu/jammy_flows
authors:
- name: Thorsten Glüsenkamp
url: https://github.com/thoglu
lang: PyTorch
description: A package that models joint (conditional) PDFs on tensor products of manifolds (Euclidean, sphere, interval, simplex) - like inverse autoregressive flows, but connects manifolds, models conditional PDFs, and allows for arbitrary couplings instead of affine ones. Includes a few SOTA flows like Gaussianization flows.
date_added: 2022-10-13
- title: normflows
date: 2020-01-28
url: https://github.com/VincentStimper/normalizing-flows
authors:
- name: Vincent Stimper
url: https://github.com/VincentStimper
lang: PyTorch
description: The library provides most of the common normalizing flow architectures. It also includes stochastic layers, flows on tori and spheres, and other tools that are particularly useful for applications to the physical sciences.
date_added: 2022-12-21
- title: ContinuousNormalizingFlows.jl
date: 2021-11-07
date_added: 2022-12-05
last_updated: 2023-05-31
url: https://github.com/impICNF/ContinuousNormalizingFlows.jl
authors:
- name: Hossein Pourbozorg
url: https://github.com/prbzrg
description: Implementations of Infinitesimal Continuous Normalizing Flows Algorithms in Julia.
lang: Julia
docs: https://impicnf.github.io/ContinuousNormalizingFlows.jl
- title: flowMC
date: 2022-06-17
date_added: 2024-06-22
last_updated: 2024-06-22
url: https://github.com/kazewong/flowMC
authors:
- name: Kaze Wong
url: https://www.kaze-wong.com/
lang: JAX
docs: https://flowmc.readthedocs.io/en/main/
description: Normalizing-flow enhanced sampling package for probabilistic inference
- title: GWKokab
date: 2024-07-05
date_added: 2024-09-21
last_updated: 2024-09-21
url: https://github.com/gwkokab/gwkokab
authors:
- name: Meesum Qazalbash
url: https://github.com/Qazalbash
- name: Muhammad Zeeshan
url: https://ccrg.rit.edu/user/muhammad.zeeshan
- name: Richard O'Shaughnessy
url: https://ccrgpages.rit.edu/~oshaughn/Richard_OShaughnessy/Home.html
lang: JAX
docs: https://gwkokab.readthedocs.io
description: A JAX-based gravitational-wave population inference toolkit for parametric models