dasayan05 / iclr24_blog_code

Related projects ⓘ

Alternatives and complementary repositories for iclr24_blog_code

• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆24Updated 4 months ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆30Updated 4 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆28Updated last year
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated last year
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated last year
• 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 4 months ago
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆27Updated 8 months ago
• srush / Tensor-Puzzles-Penzai
  ☆19Updated 7 months ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆64Updated this week
• KellerJordan / top-sgd
  Optimization algorithm which fits a ResNet to CIFAR-10 5x faster than SGD / Adam (with terrible generalization)
  ☆12Updated last year
• tanelp / confviews
  Generate bird's-eye views of conference proceedings.
  ☆22Updated 4 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆32Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• Lightning-Universe / lightning-diffusion_component
  ☆26Updated last year
• cgarciae / einop
  ☆58Updated 2 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆30Updated 4 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆82Updated 2 weeks ago
• pengyan510 / torcheck
  ☆37Updated 2 years ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆30Updated last year
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆17Updated 3 weeks ago
• NVlabs / Forecasting-Model-Search
  A system for automating selection and optimization of pre-trained models from the TAO Model Zoo
  ☆22Updated 4 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 2 years ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated 3 weeks ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆57Updated 11 months ago