Alternatives and similar repositories for maskedtensor:

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

• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated last year
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆54Updated 2 years ago
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆46Updated 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…
  ☆50Updated 2 years ago
• ag1988 / dss
  Sequence Modeling with Structured State Spaces
  ☆63Updated 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
  ☆99Updated 2 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• Lightning-Universe / paper-AAVAE
  ☆47Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆94Updated 4 months ago
• google-deepmind / einshape
  ☆102Updated 10 months ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆95Updated 3 years ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated 2 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
• proger / nanokitchen
  Parallel Associative Scan for Language Models
  ☆18Updated last year
• AranKomat / Metroplex
  ☆21Updated 2 years ago
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated last year
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆194Updated 2 months ago
• fattorib / transformer_shmap
  Tensor Parallelism with JAX + Shard Map
  ☆11Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 5 months ago
• sholtodouglas / scalingExperiments
  ☆60Updated 3 years ago
• google-research / precondition
  ☆31Updated last week
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• shikaiqiu / compute-better-spent
  ☆52Updated 6 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 5 months ago
• Newbeeer / Anytime-Auto-Regressive-Model
  Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"
  ☆26Updated last year