autonomousvision / akorn
Artificial Kuramoto Oscillatory Neurons
☆45Updated this week
Alternatives and similar repositories for akorn:
Users that are interested in akorn are comparing it to the libraries listed below
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆81Updated last year
- Comparison between GFlowNets & Maximum Entropy RL☆15Updated 10 months ago
- Flow-matching algorithms in JAX☆82Updated 5 months ago
- NF-Layers for constructing neural functionals.☆79Updated last year
- ☆20Updated 3 months ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆41Updated last year
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆20Updated last year
- Code for the paper: Rotating Features for Object Discovery☆49Updated 5 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆42Updated last month
- Modern Fixed Point Systems using Pytorch☆85Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- ☆48Updated 11 months ago
- Code for https://arxiv.org/abs/2406.04329☆51Updated last month
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated 11 months ago
- Pytorch-like dataloaders in JAX.☆67Updated 2 months ago
- ☆59Updated 3 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆54Updated 4 months ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆67Updated last week
- The Energy Transformer block, in JAX☆54Updated last year
- ☆18Updated last month
- ☆146Updated last month
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆76Updated 5 months ago
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- Code repository for Trajectory Flow Matching☆50Updated 2 months ago
- ☆49Updated 7 months ago
- ☆21Updated 5 months ago
- A modular, easy to extend GFlowNet library☆246Updated this week
- Simple illustrative examples for energy-based models in PyTorch☆61Updated 4 years ago
- ☆31Updated 7 months ago
- PyTorch Package For Quasimetric Learning☆41Updated 2 months ago