chalk-diagrams / chalk

Related projects ⓘ

Alternatives and complementary repositories for chalk

• mle-infrastructure / mle-monitor
  Lightweight Experiment & Resource Monitoring 📺
  ☆187Updated last year
• patrick-kidger / mkposters
  Make posters from Markdown files.
  ☆330Updated last year
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆175Updated 2 years ago
• google-deepmind / synjax
  ☆240Updated last month
• srush / streambook
  Live Python Notebooks with any Editor
  ☆276Updated last year
• lebrice / SimpleParsing
  Simple, Elegant, Typed Argument Parsing with argparse
  ☆432Updated 3 weeks ago
• hristo-vrigazov / mmap.ninja
  Memory mapped numpy arrays of varying shapes
  ☆285Updated 5 months ago
• google / tree-math
  Mathematical operations for JAX pytrees
  ☆189Updated 6 months ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆152Updated this week
• eladrich / pyrallis
  Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure …
  ☆198Updated 11 months ago
• google-deepmind / einshape
  ☆99Updated 4 months ago
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆337Updated 2 weeks ago
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆113Updated last year
• namedtensor / notation
  ☆108Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆215Updated this week
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆328Updated last month
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆158Updated last year
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆322Updated 7 months ago
• cgarciae / einop
  ☆58Updated 2 years ago
• xl0 / lovely-numpy
  NumPy arrays, ready for human consumption
  ☆63Updated 4 months ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆153Updated this week
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆321Updated this week
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆470Updated last year
• google / etils
  Collection of eclectic utils for python.
  ☆193Updated this week
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆246Updated last year
• srush / Autodiff-Puzzles
  ☆391Updated last month
• wilson-labs / cola
  Compositional Linear Algebra
  ☆432Updated 3 weeks ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆85Updated 9 months ago