Alternatives and similar repositories for jaxgarden

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

• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆45Updated last year
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 3 years ago
• muellerzr / import-timer
  Pragmatic approach to parsing import profiles for CI's
  ☆11Updated last year
• deep-diver / TFX-WandB
  ☆14Updated 2 years ago
• sgugger / torchdynamo-tests
  ☆19Updated 2 years ago
• cgarciae / jax-gravity
  ☆24Updated last year
• sayakpaul / count-tokens-hf-datasets
  This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…
  ☆27Updated 2 years ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆34Updated 8 months ago
• yiyixuxu / n-grammer-flax
  Implementation of N-Grammer in Flax
  ☆17Updated 2 years ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆114Updated 5 months ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆135Updated this week
• nateraw / hf-hub-lightning
  A PyTorch Lightning Callback for pushing models to the Hugging Face Hub 🤗⚡️
  ☆36Updated 3 years ago
• EleutherAI / best-download
  URL downloader supporting checkpointing and continuous checksumming.
  ☆19Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• sayakpaul / deploy-hf-tf-vision-models
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Updated 2 years ago
• CarperAI / treasure_trove
  ☆22Updated last year
• sholtodouglas / scalingExperiments
  ☆61Updated 3 years ago
• huggingface / hffs
  **ARCHIVED** Filesystem interface to 🤗 Hub
  ☆58Updated 2 years ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆85Updated last year
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• deep-diver / semantic-segmentation-ml-pipeline
  Machine Learning Pipeline for Semantic Segmentation with TensorFlow Extended (TFX) and various GCP products
  ☆96Updated 2 years ago
• soumik12345 / wandb-addons
  Weights & Biases Addons is a repository consisting of additional unitilities and community contributions for supercharging your Weights &…
  ☆23Updated last year
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆98Updated 3 months ago
• sayakpaul / parameter-ensemble-differential-evolution
  Shows how to do parameter ensembling using differential evolution.
  ☆10Updated 3 years ago
• apple / ml-hypercloning
  ☆48Updated 8 months ago
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated 2 months ago
• okarthikb / state-space-models
  ☆27Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆213Updated last year