hhaoyan / opt-einsum-torchLinks
Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.
☆15Updated 2 years ago
Alternatives and similar repositories for opt-einsum-torch
Users that are interested in opt-einsum-torch are comparing it to the libraries listed below
Sorting:
- FLOPS counter for all your GPU benchmarking needs☆13Updated 10 months ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 months ago
- A JAX library for Density Functional Theory.☆53Updated this week
- Exchange correlation functionals translated from libxc to jax☆45Updated 2 months ago
- Automatic High-Order Optimization for Tensors☆23Updated 2 years ago
- JAX exponential map normalising flows on sphere☆17Updated 4 years ago
- do the Fourier transform using the method of Padé approximants☆22Updated 7 months ago
- PySCF with auto-differentiation☆80Updated this week
- Differentiable scientific computing library☆150Updated 9 months ago
- JAX accelerated Quantum Monte Carlo☆78Updated last week
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆21Updated 3 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Updated 3 years ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated 2 months ago
- Generic PyTorch implementation of einsum that supports different semirings☆49Updated 10 months ago
- Efficient and Accurate Neural-Network Ansatz for Quantum Monte Carlo☆62Updated 6 months ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 2 years ago
- ☆70Updated 2 years ago
- Corresponding dataset and tools for the AdsorbML manuscript.☆40Updated 4 months ago
- Python library for numerical integration, interpolation, and differentiation on (molecular) grids.☆67Updated last month
- FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme☆70Updated 2 months ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆31Updated 11 months ago
- C++ and Python library for Polarizable Embedding☆22Updated 9 months ago
- Pure C implementation of e3nn☆18Updated 2 months ago
- Implementation of Forward Laplacian algorithm in JAX☆76Updated last week
- CUDA kernels for generalized matrix-multiplication in PyTorch☆83Updated 3 years ago
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.☆65Updated 9 months ago
- Differentiable Quantum Chemistry (only Differentiable Density Functional Theory and Hartree Fock at the moment)☆111Updated 3 years ago
- ESI-DCAFM-TACO-VDSP Summer School on "Machine Learning for Materials Hard and Soft"☆37Updated 2 years ago
- Mathematical Introduction to Electronic Structure Theory☆46Updated 6 years ago
- PySCF on IPU☆42Updated last year