Alternatives and similar repositories for s4-jax

Users that are interested in s4-jax are comparing it to the libraries listed below

• dsgiitr / ML-InterviewQs
  This is a question bank for practicing Machine Learning for Interviews.
  ☆33Updated 2 years ago
• DevPranjal / reproduction-review-kd
  Reproduction of the CVPR'21 paper Distilling Knowledge via Knowledge Review for the ML Reproducibility Challenge 2021
  ☆10Updated 3 years ago
• sahilg06 / Global-Wheat-Detection-AIcrowd
  Wheat detection using Faster RCNN
  ☆15Updated last year
• jiaweizzhao / ZerO-initialization
  ☆74Updated 2 years ago
• GuillaumeLeclerc / fastargs
  Python library for argument and configuration management
  ☆54Updated 2 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆28Updated last year
• AakashKumarNain / diffusion_models
  All about the fundamentals and working of Diffusion Models
  ☆158Updated 2 years ago
• lucidrains / discrete-key-value-bottleneck-pytorch
  Implementation of Discrete Key / Value Bottleneck, in Pytorch
  ☆88Updated 2 years ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆252Updated 2 years ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆83Updated 2 years ago
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• cloneofsimo / karras-power-ema-tutorial
  ☆52Updated last year
• OATML / RHO-Loss
  ☆208Updated 2 years ago
• sayakpaul / robustness-vit
  Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).
  ☆125Updated 2 years ago
• rom1504 / gpu-tester
  gpu tester detects broken and slow gpus in a cluster
  ☆70Updated 2 years ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆123Updated 11 months ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆42Updated 4 months ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆150Updated 2 years ago
• infocusp / diffusion_models
  Minimal standalone example of diffusion model
  ☆159Updated 3 years ago
• tulasiram58827 / plot_top_losses_keras
  This repo consists of code for plotting top loss images
  ☆13Updated 5 years ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆91Updated 3 years ago
• mbaradad / shaders21k
  Procedural Image Programs for Representation Learning - NeurIPS 2022
  ☆33Updated 9 months ago
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆214Updated 2 years ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 3 years ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆473Updated 3 years ago
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated 2 years ago
• stanislavfort / OpenAI_CLIP_adversarial_examples
  Developing adversarial examples and showing their semantic generalization for the OpenAI CLIP model (https://github.com/openai/CLIP)
  ☆26Updated 4 years ago