dasayan05 / iclr24_blog_code

Related projects ⓘ

Alternatives and complementary repositories for iclr24_blog_code

• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆35Updated 3 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆24Updated 3 months ago
• cgarciae / einop
  ☆58Updated 2 years ago
• tanelp / confviews
  Generate bird's-eye views of conference proceedings.
  ☆22Updated 3 months ago
• pengyan510 / torcheck
  ☆37Updated 2 years ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated last year
• attentionneuron / attentionneuron.github.io
  ☆26Updated 2 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆64Updated 3 months ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆31Updated last year
• KellerJordan / top-sgd
  Optimization algorithm which fits a ResNet to CIFAR-10 5x faster than SGD / Adam (with terrible generalization)
  ☆12Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• crypdick / timm-lr-scheduler-explorer
  A dashboard for exploring timm learning rate schedulers
  ☆18Updated last year
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• ClashLuke / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆39Updated last year
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆29Updated 4 months ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆25Updated 3 weeks ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆57Updated 11 months ago
• srush / Tensor-Puzzles-Penzai
  ☆18Updated 6 months ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated last year
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 2 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆25Updated 8 months ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆78Updated 9 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated last week
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago