cgarciae / dyn_plot

Related projects: ⓘ

• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆31Updated 3 months ago
• lucidrains / hyena-pytorch
  ☆23Updated this week
• cgarciae / einop
  ☆56Updated 2 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated 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
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆45Updated last month
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆29Updated 2 months ago
• srush / drop7
  ☆18Updated 5 months ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆29Updated last year
• Artur-Galstyan / statedict2pytree
  ☆40Updated 2 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆28Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• lucidrains / simplicial-attention
  ☆20Updated this week
• lucidrains / spatial-rmsnorm
  ☆11Updated this week
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆31Updated 2 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆23Updated 2 months ago
• radarFudan / mamba-minimal-jax
  ☆28Updated last week
• google-deepmind / spectral_ssm
  ☆25Updated 5 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆11Updated 2 months ago
• andyehrenberg / flaxlm
  ☆27Updated this week
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated 3 weeks ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆36Updated last year
• srush / Tensor-Puzzles-Penzai
  ☆17Updated 4 months ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆41Updated 3 months ago
• EleutherAI / mdl
  Minimum Description Length probing for neural network representations
  ☆15Updated 11 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆34Updated last month
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆13Updated 3 weeks ago
• PAIR-code / tiny-transformers
  ☆13Updated this week
• shikaiqiu / compute-better-spent
  ☆42Updated 3 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆28Updated last year