Alternatives and similar repositories for jaxgarden

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

• sgugger / torchdynamo-tests
  ☆19Updated 2 years ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆37Updated 2 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 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
• muellerzr / import-timer
  Pragmatic approach to parsing import profiles for CI's
  ☆11Updated 11 months ago
• AI-Hypercomputer / tpu-recipes
  ☆33Updated this week
• huggingface / hffs
  **ARCHIVED** Filesystem interface to 🤗 Hub
  ☆58Updated 2 years ago
• pytorch-tpu / transformers
  🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
  ☆17Updated 2 months ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• nateraw / huggingface-datasets-converter
  Scripts to convert datasets from various sources to Hugging Face Datasets.
  ☆57Updated 2 years ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• huggingface / olm-training
  Repo for training MLMs, CLMs, or T5-type models on the OLM pretraining data, but it should work with any hugging face text dataset.
  ☆93Updated 2 years ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆33Updated 7 months ago
• huggingface / fuego
  [WIP] A 🔥 interface for running code in the cloud
  ☆85Updated 2 years ago
• warner-benjamin / optimi
  Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
  ☆94Updated 10 months ago
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated 2 years ago
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆54Updated 11 months ago
• abheesht17 / whisper-keras
  ☆16Updated 2 years ago
• CarperAI / squeakily
  A library for squeakily cleaning and filtering language datasets.
  ☆47Updated last year
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆173Updated 10 months ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆121Updated this week
• deep-diver / complete-mlops-system-workflow
  ☆17Updated 2 years ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆34Updated 11 months ago
• sholtodouglas / scalingExperiments
  ☆59Updated 3 years ago
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆118Updated last year
• hkproj / multi-latent-attention
  ☆36Updated last week
• nateraw / hf-hub-lightning
  A PyTorch Lightning Callback for pushing models to the Hugging Face Hub 🤗⚡️
  ☆36Updated 3 years ago
• cgarciae / jax-gravity
  ☆24Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated last year
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆112Updated 4 months ago