autonomousvision / akornLinks
[ICLR'25] Artificial Kuramoto Oscillatory Neurons
☆89Updated 2 weeks ago
Alternatives and similar repositories for akorn
Users that are interested in akorn are comparing it to the libraries listed below
Sorting:
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- ☆65Updated 11 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆93Updated 3 months ago
- NF-Layers for constructing neural functionals.☆85Updated last year
- Code for the paper: Rotating Features for Object Discovery☆52Updated 9 months ago
- Modern Fixed Point Systems using Pytorch☆93Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 6 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆62Updated 2 months ago
- ☆30Updated 10 months ago
- Educational implementation of the Discrete Flow Matching paper☆88Updated 9 months ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆137Updated 4 months ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆26Updated this week
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling".☆29Updated 2 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆29Updated 11 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
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆79Updated this week
- Simple Guidance Mechanisms for Discrete Diffusion Models☆39Updated 5 months ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆89Updated 2 weeks ago
- ☆53Updated 8 months ago
- Implementation of PSGD optimizer in JAX☆33Updated 5 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control☆36Updated this week
- ☆146Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- Neural Optimal Transport with Lagrangian Costs☆56Updated 2 weeks ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆78Updated last week
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆121Updated last month
- Efficient Video Prediction via Sparsely Conditioned Flow Matching. In ICCV, 2023.☆15Updated last year
- My take on Flow Matching☆57Updated 4 months ago
- ☆185Updated 6 months ago