pytorch / maskedtensor

Related projects ⓘ

Alternatives and complementary repositories for maskedtensor

• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆52Updated last year
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆94Updated last year
• Lightning-Universe / paper-AAVAE
  ☆47Updated last year
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• lucidrains / logavgexp-torch
  Implementation of LogAvgExp for Pytorch
  ☆32Updated 2 years ago
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆26Updated 3 years ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆42Updated this week
• 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
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆71Updated 9 months ago
• sustcsonglin / gated_linear_attention_layer
  ☆31Updated 10 months ago
• fattorib / transformer_shmap
  Tensor Parallelism with JAX + Shard Map
  ☆11Updated last year
• Newbeeer / Anytime-Auto-Regressive-Model
  Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"
  ☆24Updated last year
• google-deepmind / einshape
  ☆98Updated 4 months ago
• AndPotap / einsum-search
  ☆24Updated last month
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆57Updated last year
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 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
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆25Updated 3 weeks ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆13Updated this week
• gregorbachmann / scaling_mlps
  ☆50Updated 5 months 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
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year