jax-dag-gflownet

Code for "Bayesian Structure Learning with Generative Flow Networks"

MIT License

Stars
36
View Code on GitHub View on X
Ecosystems: Python

Statistics for this project are still being loaded, please check back later.

Related Projects
minGPT-flax

GPT implementation in Flax

30 Sep 2021 16
dm-haiku

JAX-based neural network library

18 Feb 2020 2,816
jax-comln

Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)

02 Mar 2022 27
mlp-gpt-jax

A GPT, made only of MLPs, in Jax

21 May 2021 55
flax

Flax is a neural network library for JAX that is designed for flexibility.

10 Jan 2020 5,584
Deep-Flow-Guided-Video-Inpainting

pytorch implementation for "Deep Flow-Guided Video Inpainting"(CVPR'19)

26 Mar 2019 2,351
GraphGym

Platform for designing and evaluating Graph Neural Networks (GNN)

14 Oct 2020 1,683
jraph

A Graph Neural Network Library in Jax

23 Nov 2020 1,327
flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

20 Nov 2017 3,123
BSMS-GNN

22 Dec 2022 46
GDL

Code for Online Graph Dictionary Learning

03 Jun 2021 12
graph-generation

GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models

09 Nov 2017 671
Dynamic3DGaussians

17 Aug 2023 1,926
n-grammer-flax

Implementation of N-Grammer in Flax

04 Aug 2022 14
vrp-attention-jax

Reimplementation of the paper "Attention, Learn to Solve Routing Problems!" in jax/flax.

03 Oct 2021 7