Alternatives and similar repositories for dyn_plot

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

• cgarciae / einop
  ☆60Updated 3 years ago
• Artur-Galstyan / statedict2pytree
  ☆44Updated last month
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆23Updated 6 months ago
• hunar4321 / RLS-neural-net
  Recursive Leasting Squares (RLS) with Neural Network for fast learning
  ☆57Updated 2 years ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated last year
• davisyoshida / qax
  If it quacks like a tensor...
  ☆59Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated last year
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆100Updated last month
• mgrankin / minGPT
  minGPT in JAX
  ☆48Updated 3 years ago
• cgarciae / simple_flow_matching
  ☆24Updated 11 months ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆34Updated 2 years ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆37Updated 2 years ago
• SuReLI / NeurOps
  Implementations of growing and pruning in neural networks
  ☆22Updated 2 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆143Updated last year
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆37Updated 3 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆47Updated last year
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆115Updated 11 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆33Updated 2 years ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆14Updated 2 years ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆32Updated 2 weeks ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆38Updated 3 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• google-deepmind / tf2jax
  ☆118Updated last month
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 3 years ago
• davisyoshida / haiku-mup
  A port of muP to JAX/Haiku
  ☆25Updated 3 years ago