alvarobartt / safejax

Related projects ⓘ

Alternatives and complementary repositories for safejax

• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆31Updated last year
• warner-benjamin / optimi
  Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
  ☆58Updated 3 months ago
• HomebrewNLP / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆46Updated 9 months ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆95Updated 2 weeks ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 9 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated last week
• sholtodouglas / scalingExperiments
  ☆56Updated 2 years ago
• 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
• huggingface / bloom-jax-inference
  ☆64Updated 2 years ago
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆76Updated 2 years ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 5 months ago
• cat-state / tinypar
  ☆20Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆71Updated 9 months ago
• cloneofsimo / min-fsdp
  ☆72Updated 4 months ago
• HomebrewNLP / HomebrewNLP
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• j10labs / wandview
  Mobile Viewer for W&B, built on top of Flutter.
  ☆30Updated 8 months ago
• ClashLuke / HeavyBall
  Efficient optimizers
  ☆42Updated this week
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆78Updated 9 months ago
• epfml / DenseFormer
  ☆76Updated 6 months ago
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• mgmalek / efficient_cross_entropy
  ☆76Updated 5 months ago
• huggingface / peft-pytorch-conference
  Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…
  ☆14Updated last year
• cgarciae / einop
  ☆58Updated 2 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆38Updated last year
• srush / Tensor-Puzzles-Penzai
  ☆18Updated 6 months ago
• ClashLuke / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆39Updated last year
• lucidrains / memory-editable-transformer
  My explorations into editing the knowledge and memories of an attention network
  ☆34Updated last year
• cloneofsimo / zeroshampoo
  ☆31Updated 2 months ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆43Updated this week