Alternatives and similar repositories for ultra-mem

Users that are interested in ultra-mem are comparing it to the libraries listed below

• CLAIRE-Labo / EvoTune
  Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.
  ☆118Updated 3 weeks ago
• lucidrains / improving-transformers-world-model-for-rl
  Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
  ☆145Updated 7 months ago
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆169Updated last month
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆60Updated 9 months ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆102Updated 2 months ago
• lucidrains / rewind-reward-pytorch
  Implementation of ReWiND, "Language-Guided Rewards Teach Robot Policies without New Demonstrations", from USC / Amazon Robotics
  ☆35Updated 3 months ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆68Updated 2 months ago
• facebookresearch / llm-speedrunner
  The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…
  ☆112Updated 2 months ago
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆100Updated last year
• lucidrains / x-transformers-rl
  Implementation of a transformer for reinforcement learning using `x-transformers`
  ☆69Updated 2 months ago
• okarthikb / state-space-models
  ☆28Updated last year
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆99Updated last month
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆33Updated last year
• main-horse / hnet-old
  H-Net Dynamic Hierarchical Architecture
  ☆80Updated 2 months ago
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆41Updated 9 months ago
• apoorvkh / torchrunx
  Easily run PyTorch on multiple GPUs & machines
  ☆54Updated 3 weeks ago
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆194Updated last year
• codingfisch / flashrl
  Fast reinforcement learning 💨
  ☆28Updated 4 months ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆116Updated last month
• srush / mamba-primer
  ☆38Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 6 months ago
• lucidrains / openevolve
  Open-source implementation of AlphaEvolve
  ☆23Updated 6 months ago
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆57Updated last year
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆125Updated 2 months ago
• lucidrains / mind-evolution
  Implementation of Mind Evolution, Evolving Deeper LLM Thinking, from Deepmind
  ☆57Updated 6 months ago
• vdesai2014 / inference-optimization-blog-post
  ☆90Updated last year
• thinking-machines-lab / manifolds
  Supporting code for the blog post on modular manifolds.
  ☆104Updated 2 months ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆132Updated last month