Alternatives and similar repositories for JAX-examples:

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

• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆54Updated 2 years ago
• pengyan510 / torcheck
  ☆38Updated 2 years ago
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 3 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆87Updated 2 years ago
• LukeWood / reef-net
  ☆24Updated 2 years ago
• sayakpaul / ViT-jax2tf
  This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
  ☆33Updated 2 years ago
• sayakpaul / parameter-ensemble-differential-evolution
  Shows how to do parameter ensembling using differential evolution.
  ☆10Updated 3 years ago
• The-AI-Summer / RNN_tutorial
  Recurrent neural networks: building a custom LSTM/GRU cell in PyTorch
  ☆28Updated 4 years ago
• lordtt13 / transformers-experiments
  All my experiments with the various transformers and various transformer frameworks available
  ☆14Updated 3 years ago
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆32Updated 2 years ago
• PrototypeML / PrototypeML
  ☆30Updated 4 years ago
• sayakpaul / count-tokens-hf-datasets
  This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…
  ☆24Updated 2 years ago
• The-AI-Summer / simclr
  An education step by step implementation of SimCLR that accompanies the blogpost
  ☆32Updated 2 years ago
• borisdayma / lightning-kitti
  Semantic Segmentation with Pytorch-Lightning
  ☆63Updated 4 years ago
• lucidrains / feedback-transformer-pytorch
  Implementation of Feedback Transformer in Pytorch
  ☆105Updated 3 years ago
• oke-aditya / pytorch_paper_implementations
  Deep Learning Paper Implementations in PyTorch
  ☆17Updated 4 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆26Updated 6 months ago
• elliotwaite / pytorch-hooks-tutorial
  Examples of using PyTorch hooks, as covered in my YouTube tutorial video.
  ☆33Updated last year
• Yard1 / ray-skorch
  Distributed skorch on Ray Train
  ☆57Updated 2 years ago
• maelfabien / NLP_Summer_School-2021_Speech_Tutorial
  ☆11Updated 3 years ago
• rasbt / pytorch-fabric-demo
  ☆24Updated last year
• Lightning-Universe / lightning-diffusion_component
  ☆26Updated last year
• XinshaoAmosWang / DeepCriticalLearning
  Deep Critical Learning. Implementation of ProSelfLC, IMAE, DM, etc.
  ☆31Updated 2 years ago
• msaroufim / ML-devops
  Helper scripts I use to run many experiments in the morning to check at night
  ☆19Updated 3 years ago
• The-AI-Summer / learn-deep-learning
  AI Summer's complete catalog of articles
  ☆109Updated 3 years ago
• will-thompson-k / deeplearning-nlp-models
  A small, interpretable codebase containing the re-implementation of a few "deep" NLP models in PyTorch. Colab notebooks to run with GPUs.…
  ☆76Updated 4 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆29Updated 3 years ago
• abheesht17 / whisper-keras
  ☆16Updated last year
• Lightning-AI / ecosystem-ci
  Automate issue discovery for your projects against Lightning nightly and releases.
  ☆45Updated this week
• ayulockin / DataAugmentationTF
  Implementation of modern data augmentation techniques in TensorFlow 2.x to be used in your training pipeline.
  ☆34Updated 4 years ago