Alternatives and similar repositories for CS344_2021

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

• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆56Updated 3 years ago
• anonymous-code-0 / lie-transformer
  Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups
  ☆36Updated 4 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• ahennequ / cuda-tensorcores-register-mapping
  ☆19Updated 2 years ago
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆33Updated 2 years ago
• jotaf98 / shareddataset
  A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes
  ☆22Updated 3 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• lucidrains / kronecker-attention-pytorch
  Implementation of Kronecker Attention in Pytorch
  ☆19Updated 5 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• unixpickle / learn-nerf
  Learning about Neural Radiance Fields
  ☆67Updated 3 years ago
• FrancescoSaverioZuppichini / Loading-huge-PyTorch-models-with-linear-memory-consumption
  Little article showing how to load pytorch's models with linear memory consumption
  ☆34Updated 3 years ago
• osiriszjq / Rethinking-positional-encoding
  ☆89Updated last year
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆56Updated 2 years ago
• soumith / rgbd_streamer
  ☆15Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago
• sayakpaul / Adaptive-Gradient-Clipping
  Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.
  ☆85Updated 4 years ago
• utsaslab / MONeT
  MONeT framework for reducing memory consumption of DNN training
  ☆174Updated 4 years ago
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆65Updated 3 years ago
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆214Updated 2 years ago
• juntang-zhuang / ACProp-Optimizer
  Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)
  ☆16Updated 3 years ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆108Updated 2 years ago
• cpcp1998 / PermuteFormer
  Code for the paper PermuteFormer
  ☆42Updated 3 years ago
• tanelp / confviews
  Generate bird's-eye views of conference proceedings.
  ☆24Updated 2 months ago
• rwightman / efficientnet-jax
  EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
  ☆128Updated last year
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆94Updated 3 years ago
• lucidrains / compositional-attention-pytorch
  Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…
  ☆51Updated 3 years ago
• srush / torch-queue
  ☆69Updated last year
• amro-kamal / ObjectPose
  ☆13Updated 3 years ago
• rwightman / imagenet-12k
  ImageNet-12k subset of ImageNet-21k (fall11)
  ☆21Updated 2 years ago