80 Open source projects that are alternatives of or similar to score_flow

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

No description or website provided.

Python code for "Machine learning: a probabilistic perspective" (2nd edition)

Unofficial JAX implementations of deep learning research papers

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

Collecting research materials on EBM/EBL (Energy Based Models, Energy Based Learning)

KoCLIP: Korean port of OpenAI CLIP, in Flax

Chef Transformer 🍲 .

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022

JAX implementations of core Deep RL algorithms

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

Official PyTorch code for WACV 2022 paper "CFLOW-AD: Real-Time Unsupervised Anomaly Detection with Localization via Conditional Normalizing Flows"

👩 Pytorch and Jax code for the Madam optimiser.

Implementation of "Intensity-Free Learning of Temporal Point Processes" (Spotlight @ ICLR 2020)

[NeurIPS 2021 Spotlight] Aligning Pretraining for Detection via Object-Level Contrastive Learning

Implementation of Unconstrained Monotonic Neural Network and the related experiments. These architectures are particularly useful for modelling monotonic transformations in normalizing flows.

Code for the paper Semi-Conditional Normalizing Flows for Semi-Supervised Learning

The PyTorch implementation of Directed Graph Contrastive Learning (DiGCL), NeurIPS-2021

A Julia framework for invertible neural networks

Bayesian Optimization in JAX

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).

A didactic Gaussian process package for researchers in Jax.

Unsupervised Part Discovery from Contrastive Reconstruction (NeurIPS 2021)

We got a stew going!

Semi-Supervised Semantic Segmentation via Adaptive Equalization Learning, NeurIPS 2021 (Spotlight)

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

PyTorch Implementation of Google Brain's WaveGrad 2: Iterative Refinement for Text-to-Speech Synthesis

Implementation of Normalizing flows on MNIST https://arxiv.org/abs/1505.05770

A combination of VAE's and Diffusion Models for efficient, controllable and high-fidelity generation from low-dimensional latents

PIX is an image processing library in JAX, for JAX.

Code for reproducing results in the sliced score matching paper (UAI 2019)

Documentation:

PyTorch implementation of NVIDIA WaveGlow with constant memory cost.

Federated posterior averaging implemented in JAX

Continuous-time gradient flow for generative modeling and variational inference

ShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)

Multimodal Image Synthesis and Editing: A Survey

A collection of optimizers, some arcane others well known, for Flax.

A Python Library for Deep Probabilistic Modeling

JAX-based Spiking Neural Network framework

PyTorch code of "Modeling Continuous Stochastic Processes with Dynamic Normalizing Flows" (NeurIPS 2020)

Prototypical Cross-Attention Networks for Multiple Object Tracking and Segmentation, NeurIPS 2021 Spotlight

Toy implementations of some popular ML optimizers using Python/JAX

Implementations of normalizing flows using python and tensorflow

A GPT, made only of MLPs, in Jax

Code of the NeurIPS 2021 paper: Language and Visual Entity Relationship Graph for Agent Navigation

Massively parallel rigidbody physics simulation on accelerator hardware.

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

Progressive Coordinate Transforms for Monocular 3D Object Detection, NeurIPS 2021

Official implementation of CATs

Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'

Revisiting Rainbow

Fast Image Retrieval (FIRe) is an open source project to promote image retrieval research. It implements most of the major binary hashing methods to date, together with different popular backbone networks and public datasets.

Implementation of Neural Distance Embeddings for Biological Sequences (NeuroSEED) in PyTorch (NeurIPS 2021)

A Python library for machine-learning and feedback loops on streaming data

Computational Fluid Dynamics in JAX

Implementation of https://srush.github.io/annotated-s4