Ilyes Batatia's starred repositories
GPU-Puzzles
Solve puzzles. Learn CUDA.
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
matbench-discovery
An evaluation framework for machine learning models simulating high-throughput materials discovery.
torch-dftd
pytorch implementation of dftd2 & dftd3
Gaunt-Tensor-Product
[ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Products"
Awesome-Triton-Kernels
Collection of kernels written in Triton language
ACEpotentials.jl
Machine Learning Interatomic Potentials with the Atomic Cluster Expansion
mace-tutorials
Collection of tutorials to use the MACE machine learning force field.
EquivariantModels.jl
Tools for geometric learning
mace-mp-umap
UMAP analysis using a pretrained MACE model
Guaranteed-Non-Local-Molecular-Dataset
A dataset for benchmarking non-local capabilities of geometric machine learning models.