Alternatives and similar repositories for treescope:

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

• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,722Updated last month
• amakelov / mandala
  A simple & elegant experiment tracking framework that integrates persistence logic & best practices directly into Python
  ☆519Updated 2 weeks ago
• FlorianDietz / comgra
  A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…
  ☆279Updated last month
• google-deepmind / nanodo
  ☆203Updated 6 months ago
• srush / Autodiff-Puzzles
  ☆413Updated 3 months ago
• PaulPauls / llama3_interpretability_sae
  A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…
  ☆605Updated 2 months ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆219Updated 6 months ago
• BlackHC / neural_net_checklist
  ☆149Updated 5 months ago
• google-deepmind / recurrentgemma
  Open weights language model from Google DeepMind, based on Griffin.
  ☆614Updated 6 months ago
• labmlai / inspectus
  LLM Analytics
  ☆637Updated 3 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆134Updated this week
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆331Updated last year
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆344Updated 2 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆486Updated last week
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆280Updated 5 months ago
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆342Updated this week
• wilson-labs / cola
  Compositional Linear Algebra
  ☆456Updated this week
• mlecauchois / micrograd-cuda
  ☆237Updated 10 months ago
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆536Updated this week
• endia-ai / Endia
  ☆230Updated this week
• thomasahle / tensorgrad
  Tensor Network Library with Autograd
  ☆158Updated this week
• facebookresearch / searchformer
  Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".
  ☆352Updated 7 months ago
• google-deepmind / synjax
  ☆244Updated 3 months ago
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆868Updated this week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆261Updated 7 months ago
• revalo / tree-diffusion
  Diffusion on syntax trees for program synthesis
  ☆439Updated 7 months ago
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆334Updated 2 months ago
• AnswerDotAI / shell_sage
  ShellSage saves sysadmins’ sanity by solving shell script snafus super swiftly
  ☆273Updated this week
• DiscoGrad / DiscoGrad
  DiscoGrad - automatically differentiate across conditional branches in C++ programs
  ☆203Updated 4 months ago
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆256Updated last year