fabian-sp / MoMo

Related projects: ⓘ

• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆82Updated 9 months ago
• radarFudan / mamba-minimal-jax
  ☆28Updated this week
• google-deepmind / spectral_ssm
  ☆25Updated 5 months ago
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆47Updated 7 months ago
• MichaelTMatthews / Craftax_Baselines
  ☆14Updated last month
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆38Updated last year
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆40Updated last month
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆52Updated last month
• johnryan465 / pscan
  ☆33Updated 8 months ago
• shikaiqiu / compute-better-spent
  ☆42Updated 3 months ago
• i404788 / s5-pytorch
  Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)
  ☆58Updated 4 months ago
• GFNOrg / gfn-diffusion
  ☆18Updated this week
• DenisBless / variational_sampling_methods
  [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".
  ☆15Updated 3 weeks ago
• twitter-research / hyperbolic-rl
  ☆56Updated last year
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆13Updated this week
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆48Updated 9 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆31Updated 2 months ago
• MichalBortkiewicz / JaxGCRL
  ☆25Updated this week
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆17Updated 4 months ago
• chongyi-zheng / td_infonce
  ☆23Updated 10 months ago
• AllanYangZhou / universal_neural_functional
  ☆46Updated 7 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆17Updated 3 weeks ago
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆26Updated last month
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆19Updated 3 months ago
• nikhilvyas / SOAP
  ☆30Updated this week
• ZhaolinGao / REBEL
  ☆24Updated 2 weeks ago
• keraJLi / synthetic-gymnax
  ☆17Updated 3 months ago
• yilundu / ired_code_release
  ☆39Updated 3 months ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆35Updated last month
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆27Updated last week