Alternatives and similar repositories for safejax:

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

• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• HomebrewML / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆42Updated last year
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆96Updated last month
• cloneofsimo / zeroshampoo
  ☆33Updated 5 months ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆48Updated last year
• srush / drop7
  ☆18Updated 10 months ago
• cgarciae / einop
  ☆59Updated 2 years ago
• warner-benjamin / optimi
  Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
  ☆84Updated 7 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 2 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆29Updated 4 months ago
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆68Updated last year
• google-research / jestimator
  Amos optimizer with JEstimator lib.
  ☆81Updated 9 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆86Updated 2 months ago
• lucidrains / memory-editable-transformer
  My explorations into editing the knowledge and memories of an attention network
  ☆34Updated 2 years ago
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆79Updated 2 years ago
• cat-state / tinypar
  ☆20Updated last year
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 9 months ago
• google-research / precondition
  ☆31Updated last month
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated last year
• cloneofsimo / min-fsdp
  ☆75Updated 7 months ago
• sholtodouglas / scalingExperiments
  ☆58Updated 2 years ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated 5 months ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆47Updated this week
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆37Updated 10 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆76Updated last year
• ethansmith2000 / TransformerExperiments
  ☆19Updated 5 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆57Updated 3 months ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated this week
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago