autonomousvision / akorn

Related projects ⓘ

Alternatives and complementary repositories for akorn

• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 6 months ago
• augustwester / gflownet
  A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)
  ☆40Updated last year
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆76Updated last year
• kuleshov-group / awesome-discrete-diffusion-models
  A curated list for awesome discrete diffusion models resources.
  ☆61Updated this week
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆74Updated 3 months ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆86Updated 11 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• kuleshov-group / mdlm
  Simplified Masked Diffusion Language Model
  ☆202Updated this week
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆86Updated 4 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆19Updated 5 months ago
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• nnaisense / bayesian-flow-networks
  This is the official code release for Bayesian Flow Networks.
  ☆254Updated 3 months ago
• nikhilvyas / SOAP
  ☆121Updated this week
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆43Updated last year
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆231Updated this week
• yilundu / ired_code_release
  ☆46Updated 4 months ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆84Updated last month
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆96Updated last year
• lindermanlab / S5
  ☆258Updated 3 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆50Updated 10 months ago
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆120Updated last year
• louaaron / Score-Entropy-Discrete-Diffusion
  [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)
  ☆394Updated 8 months ago
• GallagherCommaJack / modulax
  ☆16Updated 2 months ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆70Updated 5 months ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆27Updated last month
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆52Updated 2 months ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆75Updated 10 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆239Updated 10 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆78Updated 9 months ago
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆48Updated last week