Alternatives and similar repositories for Yaae

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

• 3outeille / GANumpy
  A Numpy implementation of a Generative Adversarial Network.
  ☆17Updated 5 years ago
• amirabbasasadi / RockyML
  ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems
  ☆20Updated 2 years ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆41Updated last year
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 4 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• RobertTLange / code-and-blog
  Some small scale experiments for my blog posts 📝
  ☆80Updated 3 years ago
• umangjpatel / kerax
  Keras-like APIs for JAX framework
  ☆50Updated 2 years ago
• SymJAX / SymJAX
  Documentation:
  ☆127Updated 2 years ago
• sdpython / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆66Updated 2 years ago
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆159Updated 2 years ago
• fastai / fastdot
  A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges
  ☆63Updated 3 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• google-deepmind / fancyflags
  A Python library for defining structured command-line flags.
  ☆33Updated 7 months ago
• mcleonard / xaby
  Functional machine learning for fun
  ☆85Updated 4 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆211Updated last year
• sradc / SmallPebble
  A minimalist deep learning library written from scratch in Python
  ☆132Updated 3 weeks ago
• google-deepmind / tf2jax
  ☆120Updated 2 weeks ago
• rwightman / efficientnet-jax
  EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
  ☆129Updated 2 years ago
• french-ai / reinforcement
  Reinforcement learning modular with pytorch
  ☆11Updated 5 years ago
• 8bitmp3 / JAX-Flax-Tutorial-Image-Classification-with-Linen
  How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…
  ☆25Updated 2 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆110Updated last year
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆45Updated 6 months ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• sash-a / es_pytorch
  High performance implementation of Deep neuroevolution in pytorch using mpi4py. Intended for use on HPC clusters
  ☆27Updated 4 years ago
• evcu / numpy_autograd
  a simple implementation of autograd engine
  ☆24Updated 7 years ago
• rsokl / MyGrad
  Drop-in autodiff for NumPy.
  ☆213Updated last month
• LukeWood / keras-genetic
  ☆18Updated 3 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆32Updated 5 years ago
• ofnote / tsanley
  A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")
  ☆40Updated 6 years ago