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 4 years ago
• umangjpatel / kerax
  Keras-like APIs for JAX framework
  ☆50Updated 2 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 3 years ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 3 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated 11 months ago
• AdilZouitine / paper_notes
  [WIP] Note on the research papers I read
  ☆7Updated 5 years ago
• AdilZouitine / outfit
  Tidy up your machine learning experiments
  ☆17Updated 5 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• cgarciae / jax-gravity
  ☆24Updated last year
• 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…
  ☆24Updated last year
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆213Updated 2 years ago
• theblackcat102 / H5Record
  Large dataset storage format for Pytorch
  ☆45Updated 3 years ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 3 years ago
• SymJAX / SymJAX
  Documentation:
  ☆119Updated last year
• sdpython / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆65Updated last year
• french-ai / reinforcement
  Reinforcement learning modular with pytorch
  ☆11Updated 4 years ago
• 3outeille / CNNumpy
  A Numpy implementation of a Convolutional Neural Network: slow & fast (im2col/col2im).
  ☆56Updated last year
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆158Updated last year
• sash-a / es_pytorch
  High performance implementation of Deep neuroevolution in pytorch using mpi4py. Intended for use on HPC clusters
  ☆26Updated 3 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆71Updated 3 years ago
• fastai / fastdot
  A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges
  ☆63Updated 2 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated last year
• hardmaru / mdn_jax_tutorial
  Mixture Density Networks (Bishop, 1994) tutorial in JAX
  ☆59Updated 4 years ago
• cair / PyTsetlinMachineCUDA
  Massively Parallel and Asynchronous Architecture for Logic-based AI
  ☆41Updated 2 years ago
• mle-infrastructure / mle-logging
  Lightweight ML Experiment Logging 📖
  ☆80Updated 7 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆193Updated 2 months ago
• shawwn / ml-notes
  ☆39Updated 2 years ago