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 5 months ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆112Updated last week
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 2 years ago
• google-research / precondition
  ☆31Updated 4 months ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆59Updated 3 weeks 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
• AI-Hypercomputer / tpu-recipes
  ☆51Updated last week
• warner-benjamin / optimi
  Fast, Modern, and Low Precision PyTorch Optimizers
  ☆116Updated 2 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• erfanzar / eformer
  (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX
  ☆28Updated 2 weeks ago
• VikParuchuri / triton_tutorial
  Tutorials for Triton, a language for writing gpu kernels
  ☆56Updated 2 years ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆96Updated 3 months ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆34Updated last year
• BlackHC / neural_net_checklist
  ☆150Updated last year
• muellerzr / nbdistributed
  Seemless interface of using PyTOrch distributed with Jupyter notebooks
  ☆53Updated last month
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆103Updated 10 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆64Updated last year
• srush / Tensor-Puzzles-Penzai
  ☆21Updated last year
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆48Updated last year
• SeunghyunSEO / optimized_hf_llama_class_for_training
  ☆48Updated last year
• apple / ml-hypercloning
  ☆52Updated 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
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆86Updated 3 years ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆25Updated 8 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆39Updated 2 months ago
• epfml / DenseFormer
  ☆81Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• apple / ml-ademamix
  ☆68Updated 11 months ago