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
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• lucidrains / kronecker-attention-pytorch
  Implementation of Kronecker Attention in Pytorch
  ☆19Updated 4 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• anonymous-code-0 / lie-transformer
  Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups
  ☆36Updated 4 years ago
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆32Updated 2 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
• tanelp / confviews
  Generate bird's-eye views of conference proceedings.
  ☆24Updated 6 months ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated 2 years ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• unixpickle / learn-nerf
  Learning about Neural Radiance Fields
  ☆67Updated 3 years ago
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆45Updated 11 months ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆54Updated 2 years ago
• jotaf98 / shareddataset
  A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes
  ☆20Updated 3 years ago
• juntang-zhuang / ACProp-Optimizer
  Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)
  ☆15Updated 3 years ago
• stanford-futuredata / loa
  Public code for LOA
  ☆19Updated 2 years ago
• Lightning-AI / forked-pdb
  Python pdb for multiple processes
  ☆49Updated last month
• lucidrains / remixer-pytorch
  Implementation of the Remixer Block from the Remixer paper, in Pytorch
  ☆36Updated 3 years ago
• srush / torch-queue
  ☆68Updated last year
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated 2 years ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆56Updated last week
• cat-state / clip_benchmark
  clip retrieval benchmark
  ☆17Updated 3 years ago
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆38Updated 3 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 4 years ago
• Dao-AILab / gemm-cublas
  ☆21Updated last month
• lucidrains / deep-linear-network
  A simple implementation of a deep linear Pytorch module
  ☆21Updated 4 years ago