Alternatives and similar repositories for Learning-JAX

Users that are interested in Learning-JAX are comparing it to the libraries listed below

• ml-gde / jaxgarden
  A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax
  ☆21Updated 3 months ago
• apoorvkh / torchrunx
  Easily run PyTorch on multiple GPUs & machines
  ☆47Updated 2 months ago
• SeunghyunSEO / optimized_hf_llama_class_for_training
  ☆48Updated last year
• devvrit / matformer
  MatFormer repo
  ☆62Updated 9 months ago
• apple / ml-hypercloning
  ☆50Updated 10 months ago
• NX-AI / mlstm_kernels
  Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.
  ☆68Updated last month
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆60Updated 2 weeks ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 2 years ago
• Aleph-Alpha-Research / trigrams
  ☆58Updated 4 months ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆48Updated last year
• sayakpaul / deploy-hf-tf-vision-models
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Updated 3 years ago
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆107Updated 5 months ago
• huggingface / competitions
  ☆124Updated 10 months ago
• microsoft / encoder-decoder-slm
  Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…
  ☆29Updated 7 months ago
• Knowledgator / FlashDeBERTa
  Trully flash implementation of DeBERTa disentangled attention mechanism.
  ☆63Updated last week
• epfml / DenseFormer
  ☆82Updated last year
• Upaya07 / NeurIPS-llm-efficiency-challenge
  Code for NeurIPS LLM Efficiency Challenge
  ☆59Updated last year
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆109Updated last week
• huggingface / chug
  Minimal sharded dataset loaders, decoders, and utils for multi-modal document, image, and text datasets.
  ☆158Updated last year
• kevinwu23 / StanfordFineTuneBench
  ☆31Updated 10 months ago
• IlyasMoutawwakil / py-txi
  A Python wrapper around HuggingFace's TGI (text-generation-inference) and TEI (text-embedding-inference) servers.
  ☆33Updated last week
• NathanGodey / headless-lm
  Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…
  ☆27Updated last year
• muellerzr / nbdistributed
  Seemless interface of using PyTOrch distributed with Jupyter notebooks
  ☆49Updated last week
• wolfecameron / lora_instruction_tune
  ☆40Updated last year
• ariG23498 / quantized-diffusion-inference
  Notebook and Scripts that showcase running quantized diffusion models on consumer GPUs
  ☆38Updated 10 months ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆120Updated 7 months ago
• rasbt / pytorch-memory-optim
  This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…
  ☆92Updated 2 years ago
• google-research-datasets / QAmeleon
  QAmeleon introduces synthetic multilingual QA data using PaLM, a 540B large language model. This dataset was generated by prompt tuning P…
  ☆34Updated 2 years ago
• ariG23498 / fine-tune-paligemma
  Notebooks for fine tuning pali gemma
  ☆116Updated 5 months ago
• lhoestq / hfjobs
  Hugging Face Jobs
  ☆19Updated 2 months ago