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 6 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆64Updated 2 years ago
• AI-Hypercomputer / tpu-recipes
  ☆54Updated last week
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• machine-learning-apps / wandb-action
  GitHub Action That Retrieves Model Runs From Weights & Biases
  ☆60Updated 3 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆47Updated last year
• pytorch-tpu / examples
  This repository contains example code to build models on TPUs
  ☆30Updated 2 years ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆151Updated 3 months ago
• muellerzr / nbdistributed
  Seemless interface of using PyTOrch distributed with Jupyter notebooks
  ☆57Updated 2 months ago
• teddykoker / tinyloader
  ☆68Updated 8 months ago
• google-research / precondition
  ☆31Updated 3 weeks ago
• srush / Tensor-Puzzles-Penzai
  ☆21Updated last year
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆26Updated 2 years ago
• NathanGodey / headless-lm
  Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…
  ☆28Updated last year
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆116Updated 3 months ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• d-kleine / NER_decoder
  Named Entity Recognition with an decoder-only (autoregressive) LLM using HuggingFace
  ☆45Updated 2 months ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆176Updated this week
• sgugger / hf_examples
  NLP Examples using the 🤗 libraries
  ☆40Updated 4 years ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆297Updated last year
• nateraw / huggingface-datasets-converter
  Scripts to convert datasets from various sources to Hugging Face Datasets.
  ☆57Updated 3 years ago
• rwightman / genalog
  Genalog is an open source, cross-platform python package allowing generation of synthetic document images with custom degradations and te…
  ☆44Updated last year
• rasbt / datapipes-blog
  Code for the DataPipes article
  ☆16Updated 3 years ago
• stephantul / pynife
  Nearly Inference Free Embeddings: make your RAG queries 500x faster
  ☆64Updated 3 weeks ago
• BlackHC / neural_net_checklist
  ☆211Updated 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
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆100Updated last month
• nateraw / spaces-docker-templates
  🚀🤗 A collection of templates for Hugging Face Spaces
  ☆35Updated 2 years ago
• Lightning-AI / ecosystem-ci
  Automate issue discovery for your projects against Lightning nightly and releases.
  ☆46Updated 7 months ago