sholtodouglas / minformer

Related projects: ⓘ

• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆100Updated 2 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆53Updated 9 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆129Updated 2 months ago
• Reytuag / transformerXL_PPO_JAX
  ☆59Updated last month
• keraJLi / rejax
  ☆141Updated 2 weeks ago
• world-modelz / dreamax
  A scalable Dreamer implementation in JAX
  ☆11Updated 2 years ago
• neoneye / ARC-Interactive-History-Dataset
  The history files when recording human interaction while solving ARC tasks
  ☆91Updated this week
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆165Updated 3 weeks ago
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆106Updated last year
• okarthikb / state-space-models
  ☆27Updated 2 months ago
• sholtodouglas / scalingExperiments
  ☆56Updated 2 years ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code
  ☆20Updated 3 weeks ago
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆95Updated 2 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆17Updated 3 weeks ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆102Updated 3 weeks ago
• goncalog / ai-robotics
  AI Robotics tutorials for hobbyists
  ☆32Updated last month
• DramaCow / jaxued
  ☆56Updated last month
• michaelhodel / re-arc
  Reverse Engineering the Abstraction and Reasoning Corpus
  ☆130Updated last month
• facebookresearch / motif
  Intrinsic Motivation from Artificial Intelligence Feedback
  ☆117Updated 10 months ago
• RobertTLange / gymnax-blines
  Baselines for gymnax 🤖
  ☆57Updated last year
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆102Updated last month
• google-deepmind / nanodo
  ☆180Updated 2 months ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆191Updated 11 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆116Updated last month
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆87Updated last month
• jennyzzt / awesome-open-ended
  Awesome Open-ended AI
  ☆168Updated 2 months ago
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆102Updated 10 months ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆52Updated last year
• jbloomAus / DecisionTransformerInterpretability
  Interpreting how transformers simulate agents performing RL tasks
  ☆62Updated 10 months ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆59Updated last month