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
  ☆20Updated 2 months ago
• apple / ml-hypercloning
  ☆49Updated 9 months ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 2 years ago
• apoorvkh / torchrunx
  Easily run PyTorch on multiple GPUs & machines
  ☆46Updated last month
• SeunghyunSEO / optimized_hf_llama_class_for_training
  ☆48Updated 11 months ago
• NathanGodey / headless-lm
  Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…
  ☆27Updated last year
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆108Updated 3 weeks ago
• epfml / DenseFormer
  ☆82Updated 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
• Knowledgator / FlashDeBERTa
  Trully flash implementation of DeBERTa disentangled attention mechanism.
  ☆63Updated last week
• Aleph-Alpha-Research / trigrams
  ☆56Updated 3 months ago
• huggingface / competitions
  ☆124Updated 9 months ago
• microsoft / encoder-decoder-slm
  Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…
  ☆29Updated 6 months ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆87Updated last year
• kevinwu23 / StanfordFineTuneBench
  ☆31Updated 9 months ago
• LAGoM-NLP / transtokenizer
  ☆51Updated 6 months ago
• luyug / magix
  Supercharge huggingface transformers with model parallelism.
  ☆77Updated last month
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆82Updated 3 years ago
• thomwolf / sesame-explorations
  ☆29Updated 3 months ago
• devvrit / matformer
  MatFormer repo
  ☆60Updated 8 months 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
• Upaya07 / NeurIPS-llm-efficiency-challenge
  Code for NeurIPS LLM Efficiency Challenge
  ☆59Updated last year
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆107Updated 5 months ago
• TristanThrush / i-am-a-strange-dataset
  Repository for "I am a Strange Dataset: Metalinguistic Tests for Language Models"
  ☆44Updated last year
• MadryLab / platinum-benchmarks
  ☆31Updated 4 months ago
• NX-AI / mlstm_kernels
  Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.
  ☆69Updated last week
• sgugger / torchdynamo-tests
  ☆19Updated 2 years ago
• chandar-lab / NeoBERT
  ☆80Updated 2 months ago
• orevaahia / magnet-tokenization
  ☆13Updated 8 months ago
• huggingface / chug
  Minimal sharded dataset loaders, decoders, and utils for multi-modal document, image, and text datasets.
  ☆158Updated last year