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
  ☆53Updated 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
• anonymous-code-0 / lie-transformer
  Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups
  ☆36Updated 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 …
  ☆63Updated 2 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆76Updated last year
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆53Updated 2 years ago
• lucidrains / kronecker-attention-pytorch
  Implementation of Kronecker Attention in Pytorch
  ☆18Updated 4 years ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆85Updated last year
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆107Updated 2 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆71Updated 3 years ago
• cpcp1998 / PermuteFormer
  Code for the paper PermuteFormer
  ☆42Updated 3 years ago
• srush / torch-queue
  ☆68Updated last year
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆32Updated 2 years ago
• soumik12345 / nerf.jax
  A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.
  ☆13Updated 2 years ago
• rwightman / imagenet-12k
  ImageNet-12k subset of ImageNet-21k (fall11)
  ☆21Updated last year
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 5 months ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆18Updated last year
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated last year
• pengyan510 / torcheck
  ☆38Updated 2 years ago
• unixpickle / learn-nerf
  Learning about Neural Radiance Fields
  ☆67Updated 2 years ago
• akshayka / hessian_trace_estimation
  Hessian trace estimation using PyTorch and Hutch++
  ☆19Updated 4 years ago
• learning-at-home / lean_transformer
  Memory-efficient transformer. Work in progress.
  ☆19Updated 2 years ago
• tanelp / confviews
  Generate bird's-eye views of conference proceedings.
  ☆23Updated 3 months ago
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 4 years ago
• borisdayma / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆56Updated 2 years ago