Alternatives and similar repositories for maskedtensor

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

• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated 2 years ago
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆54Updated 2 years ago
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆100Updated 2 years ago
• rom1504 / gpu-tester
  gpu tester detects broken and slow gpus in a cluster
  ☆70Updated 2 years ago
• ronghanghu / vit_10b_fsdp_example
  See details in https://github.com/pytorch/xla/blob/r1.12/torch_xla/distributed/fsdp/README.md
  ☆24Updated 2 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆38Updated 3 years ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• lucidrains / logavgexp-torch
  Implementation of LogAvgExp for Pytorch
  ☆36Updated 2 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆91Updated 3 years ago
• tmbdev-archive / webdataset-lightning
  A small demonstration of using WebDataset with ImageNet and PyTorch Lightning
  ☆74Updated last year
• google-deepmind / einshape
  ☆104Updated last year
• sholtodouglas / scalingExperiments
  ☆60Updated 3 years ago
• shikaiqiu / compute-better-spent
  ☆53Updated 8 months ago
• cpcp1998 / PermuteFormer
  Code for the paper PermuteFormer
  ☆42Updated 3 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• proger / nanokitchen
  Parallel Associative Scan for Language Models
  ☆18Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• lucidrains / memory-editable-transformer
  My explorations into editing the knowledge and memories of an attention network
  ☆35Updated 2 years ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• arogozhnikov / adamw_bfloat16
  AdamW optimizer for bfloat16 models in pytorch 🔥.
  ☆32Updated last year
• lucidrains / coordinate-descent-attention
  Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk
  ☆46Updated last year
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆69Updated 2 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 11 months ago
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆46Updated 4 years ago
• google-research / precondition
  ☆31Updated last week
• 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
• google-research / jestimator
  Amos optimizer with JEstimator lib.
  ☆82Updated last year