Alternatives and similar repositories for comgra:

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

• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆334Updated last year
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆393Updated 3 weeks ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆359Updated last month
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆219Updated 8 months ago
• srush / Autodiff-Puzzles
  ☆418Updated 4 months ago
• google-deepmind / nanodo
  ☆212Updated 7 months ago
• tysam-code / hlb-gpt
  Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…
  ☆343Updated 7 months ago
• apple / ml-sigma-reparam
  ☆301Updated 8 months ago
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆282Updated 6 months ago
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆554Updated this week
• google-deepmind / recurrentgemma
  Open weights language model from Google DeepMind, based on Griffin.
  ☆623Updated 3 weeks ago
• modula-systems / modula
  🧱 Modula software package
  ☆169Updated this week
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆882Updated 10 months ago
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆891Updated this week
• RobertRiachi / nanoPALM
  ☆143Updated last year
• warner-benjamin / commented-transformers
  Highly commented implementations of Transformers in PyTorch
  ☆132Updated last year
• johnmarktaylor91 / torchlens
  Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
  ☆554Updated last week
• BlackHC / neural_net_checklist
  ☆149Updated 6 months ago
• HazyResearch / H3
  Language Modeling with the H3 State Space Model
  ☆515Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆498Updated 2 weeks 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…
  ☆603Updated 3 months ago
• sumo43 / loopvlm
  run paligemma in real time
  ☆131Updated 9 months ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆369Updated this week
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆259Updated last year
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆224Updated this week
• epistoteles / TensorHue
  TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…
  ☆115Updated 2 weeks ago
• google-deepmind / synjax
  ☆245Updated 5 months ago
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆91Updated 3 months ago
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆124Updated last year
• joennlae / tensorli
  Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).
  ☆250Updated last year