autonomousvision / akorn
[ICLR'25] Artificial Kuramoto Oscillatory Neurons
☆78Updated 2 months ago
Alternatives and similar repositories for akorn:
Users that are interested in akorn are comparing it to the libraries listed below
- Code for the paper: Rotating Features for Object Discovery☆51Updated 8 months ago
- Modern Fixed Point Systems using Pytorch☆89Updated last year
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆78Updated 2 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- NF-Layers for constructing neural functionals.☆84Updated last year
- ☆64Updated 10 months ago
- The Energy Transformer block, in JAX☆57Updated last year
- ☆52Updated 7 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆42Updated last year
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆33Updated 8 months ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 2 years ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆15Updated 3 months ago
- Educational implementation of the Discrete Flow Matching paper☆87Updated 8 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 5 months ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆78Updated 3 weeks ago
- ☆30Updated last month
- My take on Flow Matching☆52Updated 3 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆75Updated 4 months ago
- Comparison between GFlowNets & Maximum Entropy RL☆16Updated last year
- ☆29Updated 9 months ago
- ☆32Updated 11 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆79Updated 9 months ago
- A State-Space Model with Rational Transfer Function Representation.☆78Updated 11 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆123Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last month
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆61Updated 2 years ago
- ☆142Updated last year