Alternatives and similar repositories for dyn_plot:

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

• Artur-Galstyan / statedict2pytree
  ☆42Updated 3 weeks ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• cgarciae / simple_flow_matching
  ☆18Updated 3 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 9 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 10 months ago
• ethansmith2000 / TransformerExperiments
  ☆19Updated 2 weeks ago
• cloneofsimo / zeroshampoo
  ☆33Updated 7 months ago
• GallagherCommaJack / modulax
  ☆17Updated 7 months ago
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated 10 months ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆24Updated last month
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆30Updated 5 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆23Updated 2 months ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆41Updated 2 years ago
• google-deepmind / spectral_ssm
  ☆31Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 3 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 8 months ago
• mgrankin / minGPT
  minGPT in JAX
  ☆48Updated 3 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆57Updated 5 months ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆24Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 6 months ago
• google-deepmind / einshape
  ☆101Updated 9 months ago
• Fchaubard / gradient_agreement_filtering
  This is the official repo for Gradient Agreement Filtering (GAF).
  ☆23Updated 2 months ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 2 months ago