mzguntalan / zephyr

Related projects ⓘ

Alternatives and complementary repositories for zephyr

• packquickly / schedule_free_optx
  Schedule free optimiser implemented in JAX using Optimistix
  ☆14Updated 5 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• okarthikb / state-space-models
  ☆27Updated 4 months ago
• lucidrains / dreamerv3-pytorch
  Implementation of DreamerV3 in Pytorch
  ☆33Updated this week
• geohot / dumbrl
  Can RL solve simple problems?
  ☆51Updated 10 months ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆22Updated last week
• lowrollr / turbozero
  fast + parallel AlphaZero in JAX
  ☆84Updated 7 months ago
• idoh / fast_mamba.np
  A pure and fast NumPy implementation of Mamba with cache support.
  ☆17Updated 5 months ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code
  ☆27Updated this week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆200Updated last year
• theOGognf / rl8
  A high throughput, end-to-end RL library for infinite horizon tasks.
  ☆18Updated 5 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆85Updated 2 months ago
• radarFudan / mamba-minimal-jax
  ☆29Updated 2 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆82Updated 2 weeks ago
• nreHieW / r-nn
  Tensor library with autograd using only Rust's standard library
  ☆62Updated 4 months ago
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆119Updated 3 months ago
• mikelma / craftium
  A framework for creating rich, 3D, Minecraft-like environments for AI research based on Minetest
  ☆34Updated this week
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX
  ☆71Updated this week
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated last month
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆121Updated 3 months ago
• ClashLuke / HeavyBall
  Efficient optimizers
  ☆80Updated this week
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆87Updated 11 months ago
• Reytuag / transformerXL_PPO_JAX
  ☆65Updated 2 weeks ago
• swairshah / Intensify
  coloring terminal text with intensities (used for plotting probability, entropy with tokens)
  ☆12Updated last month
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 9 months ago
• keraJLi / rejax
  ☆149Updated this week
• ml-jku / DIffUCO
  ☆30Updated 2 weeks ago
• nikhilvyas / SOAP
  ☆128Updated last week