autonomousvision / akornLinks
[ICLR'25] Artificial Kuramoto Oscillatory Neurons
☆95Updated last month
Alternatives and similar repositories for akorn
Users that are interested in akorn are comparing it to the libraries listed below
Sorting:
- NF-Layers for constructing neural functionals.☆87Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆118Updated 5 months ago
- Modern Fixed Point Systems using Pytorch☆102Updated last year
- ☆67Updated last year
- Code for the paper: Rotating Features for Object Discovery☆53Updated 11 months ago
- Flow-matching algorithms in JAX☆100Updated 11 months ago
- ☆115Updated last month
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆28Updated 2 months ago
- ☆32Updated last year
- ☆38Updated 3 weeks ago
- My take on Flow Matching☆69Updated 6 months ago
- ☆65Updated 3 years ago
- ☆298Updated 6 months ago
- Implementation of PSGD optimizer in JAX☆34Updated 7 months ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- ☆51Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆93Updated 4 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆105Updated 10 months ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆82Updated 3 weeks ago
- 📄Small Batch Size Training for Language Models☆36Updated last week
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆32Updated last year
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- ☆158Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning☆400Updated 2 weeks ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆21Updated last year
- ☆35Updated 4 months ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆111Updated last week
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆54Updated 8 months ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆17Updated 6 months ago