Alternatives and similar repositories for pyfra:

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

• shawwn / ml-notes
  ☆39Updated 2 years ago
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆238Updated last year
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆69Updated 2 years ago
• lucidrains / g-mlp-gpt
  GPT, but made only out of MLPs
  ☆88Updated 3 years ago
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆187Updated 2 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆57Updated 3 years ago
• srush / parallax
  ☆153Updated 4 years ago
• shawwn / tpunicorn
  Babysit your preemptible TPUs
  ☆85Updated 2 years ago
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆142Updated 2 years ago
• EleutherAI / project-menu
  See the issue board for the current status of active and prospective projects!
  ☆65Updated 3 years ago
• srush / awesome-ml-tracking
  ☆103Updated 4 years ago
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆37Updated 3 years ago
• EleutherAI / magiCARP
  One stop shop for all things carp
  ☆59Updated 2 years ago
• multimodalart / latent-diffusion-notebook
  ☆57Updated 2 years ago
• rom1504 / gpu-tester
  gpu tester detects broken and slow gpus in a cluster
  ☆70Updated 2 years ago
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated last year
• nestordemeure / flaxOptimizers
  A collection of optimizers, some arcane others well known, for Flax.
  ☆29Updated 3 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• huggingface / bloom-jax-inference
  ☆67Updated 2 years ago
• openai / LHOPT
  Learned Hyperparameter Optimizers
  ☆59Updated 3 years ago
• lf1-io / padl
  Functional deep learning
  ☆108Updated 2 years ago
• tensorfork / OBST
  Your fruity companion for transformers
  ☆14Updated 2 years ago
• kingoflolz / cc_img_dl
  ☆27Updated 4 years ago
• fastai / fastgpu
  A queue service for quickly developing scripts that use all your GPUs efficiently
  ☆83Updated 2 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• sholtodouglas / scalingExperiments
  ☆60Updated 3 years ago
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆32Updated 2 years ago
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆105Updated 4 years ago
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆177Updated this week
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago