Alternatives and similar repositories for opt-einsum-torch

Users that are interested in opt-einsum-torch are comparing it to the libraries listed below

• xitorch / xitorch
  Differentiable scientific computing library
  ☆156Updated last year
• ReiherGroup / CoRe_optimizer
  Continual Resilient (CoRe) Optimizer for PyTorch
  ☆11Updated last year
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆66Updated 2 months ago
• BBuf / flash-rwkv
  ☆32Updated last year
• enp1s0 / ozIMMU
  FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme
  ☆96Updated last week
• bytedance / jaqmc
  JAX accelerated Quantum Monte Carlo
  ☆88Updated 4 months ago
• bytedance / LapNet
  Efficient and Accurate Neural-Network Ansatz for Quantum Monte Carlo
  ☆72Updated last year
• harvardnlp / genbmm
  CUDA kernels for generalized matrix-multiplication in PyTorch
  ☆85Updated 4 years ago
• bdusell / semiring-einsum
  Generic PyTorch implementation of einsum that supports different semirings
  ☆51Updated last 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
  ☆47Updated 2 years ago
• facebookresearch / LA-MCTS
  High dimensional black-box optimizer using Latent Action Monte Carlo Tree Search algorithm
  ☆29Updated 3 years ago
• sail-sg / d4ft
  A JAX library for Density Functional Theory.
  ☆54Updated 2 weeks ago
• facebookresearch / any4
  Quantize transformers to any learned arbitrary 4-bit numeric format
  ☆49Updated 5 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last year
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆59Updated 2 years ago
• NVIDIA / dllogger
  A logging tool for deep learning.
  ☆63Updated 8 months ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆27Updated this week
• hazan-lab / flash-stu
  PyTorch implementation of the Flash Spectral Transform Unit.
  ☆20Updated last year
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆57Updated 3 years ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• KhrulkovV / tt-pytorch
  ☆59Updated 5 years ago
• deepmodeling / openlam
  ☆21Updated 10 months ago
• arogozhnikov / adamw_bfloat16
  AdamW optimizer for bfloat16 models in pytorch 🔥.
  ☆38Updated last year
• microsoft / folx
  Implementation of Forward Laplacian algorithm in JAX
  ☆92Updated 2 weeks ago
• Aleph-Alpha-Research / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆22Updated last year
• YWolfeee / lapjax
  A JAX based package designed for efficient second order operators (e.g., laplacian) computation.
  ☆74Updated last year
• pytorch / maskedtensor
  MaskedTensors for PyTorch
  ☆38Updated 3 years ago
• tjyuyao / cutex
  PyCUDA based PyTorch Extension Made Easy
  ☆26Updated last year
• tencent-quantum-lab / TenCirChem
  Quantum computational chemistry based on TensorCircuit
  ☆70Updated 5 months ago
• CCQC / Quax
  Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.
  ☆70Updated last year