Alternatives and similar repositories for iceberg:

Users that are interested in iceberg are comparing it to the libraries listed below

• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆393Updated 3 weeks ago
• modula-systems / modula
  🧱 Modula software package
  ☆167Updated this week
• revalo / tree-diffusion
  Diffusion on syntax trees for program synthesis
  ☆446Updated 8 months ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆171Updated this week
• brentyi / tyro
  CLI interfaces & config objects, from types
  ☆583Updated this week
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆67Updated 3 months ago
• FLAIROx / Kinetix
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
  ☆167Updated this week
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆132Updated 7 months ago
• chalk-diagrams / chalk
  A declarative drawing API in Python
  ☆294Updated 6 months ago
• keraJLi / rejax
  ☆191Updated 2 months ago
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆288Updated 2 weeks ago
• google-deepmind / synjax
  ☆245Updated 5 months ago
• facebookresearch / searchformer
  Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".
  ☆362Updated 9 months ago
• google-deepmind / nanodo
  ☆212Updated 7 months ago
• DramaCow / jaxued
  ☆74Updated 6 months ago
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆47Updated last year
• yashbonde / rasp
  Implementing RASP transformer programming language https://arxiv.org/pdf/2106.06981.pdf.
  ☆50Updated 3 years ago
• lowrollr / turbozero
  fast + parallel AlphaZero in JAX
  ☆92Updated 2 months ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆49Updated 5 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆147Updated 2 weeks ago
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆198Updated last week
• Avalon-Benchmark / avalon
  A 3D video game environment and benchmark designed from scratch for reinforcement learning research
  ☆185Updated last year
• eladrich / pyrallis
  Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure …
  ☆222Updated last year
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆147Updated 8 months ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆141Updated 3 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 7 months ago
• cgarciae / einop
  ☆59Updated 3 years ago