mitkotak / fast_flopsLinks
FLOPS counter for all your GPU benchmarking needs
☆14Updated last year
Alternatives and similar repositories for fast_flops
Users that are interested in fast_flops are comparing it to the libraries listed below
Sorting:
- Pure C implementation of e3nn☆20Updated 4 months ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆62Updated this week
- Implementation of the Euclidean fast attention (EFA) algorithm☆54Updated this week
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆64Updated 9 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆40Updated 2 years ago
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆15Updated this week
- [TMLR 2024] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields☆38Updated 5 months ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆272Updated this week
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆14Updated 3 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆109Updated 4 months ago
- Equivariant machine learning interatomic potentials in JAX.☆74Updated 3 months ago
- ☆12Updated last year
- ☆33Updated 5 months ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆83Updated this week
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆23Updated 5 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆36Updated 5 months ago
- ☆22Updated 3 months ago
- PySCF on IPU☆42Updated last year
- MESS: Modern Electronic Structure Simulations☆34Updated this week
- C++/CUDA library for SO(3) equivariant operations☆25Updated this week
- Build neural networks for machine learning force fields with JAX☆123Updated 2 months ago
- ☆23Updated last year
- Code for “From Molecules to Materials Pre-training Large Generalizable Models for Atomic Property Prediction”.☆59Updated 9 months ago
- ☆18Updated 2 months ago
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆247Updated 2 months ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆52Updated 4 months ago
- ☆12Updated last year
- CUDA implementations of MACE models☆15Updated 6 months ago
- Higher-order equivariant neural networks for charge density prediction in materials☆61Updated 5 months ago
- Exchange correlation functionals translated from libxc to jax☆45Updated 4 months ago