Alternatives and similar repositories for comgra:

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

• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆402Updated 3 weeks ago
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆339Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆505Updated 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…
  ☆114Updated last month
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆365Updated this week
• srush / Autodiff-Puzzles
  ☆428Updated 5 months ago
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆285Updated 7 months ago
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆297Updated 9 months ago
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆349Updated last week
• jxbz / agd
  Automatic gradient descent
  ☆207Updated last year
• BlackHC / neural_net_checklist
  ☆150Updated 7 months ago
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆884Updated 11 months ago
• johnmarktaylor91 / torchlens
  Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
  ☆572Updated last month
• RobertRiachi / nanoPALM
  ☆143Updated 2 years ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆234Updated this week
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆221Updated 9 months ago
• google-deepmind / nanodo
  ☆215Updated 8 months ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆320Updated last year
• wilson-labs / cola
  Compositional Linear Algebra
  ☆470Updated last week
• google / autobound
  AutoBound automatically computes upper and lower bounds on functions.
  ☆358Updated 9 months ago
• eduardoleao052 / Autograd-from-scratch
  Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.
  ☆111Updated last year
• google-deepmind / synjax
  ☆246Updated 6 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…
  ☆342Updated 8 months ago
• joennlae / tensorli
  Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).
  ☆250Updated last year
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆480Updated last year
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆565Updated this week
• pbelcak / fastfeedforward
  A repository for log-time feedforward networks
  ☆221Updated last year
• apple / ml-sigma-reparam
  ☆302Updated 9 months ago
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆905Updated last week
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year