GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU
☆24Mar 27, 2025Updated 11 months ago
Alternatives and similar repositories for GeoT
Users that are interested in GeoT are comparing it to the libraries listed below
Sorting:
- "Not too complicated" training code for CIFAR-10 by PyTorch Lightning☆12Jun 5, 2022Updated 3 years ago
- PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity☆121Updated this week
- Koishi's Day 2025 Paper (NeurIPS 2025): "Codifying Character Logic in Role-Playing"☆23Jan 15, 2026Updated 2 months ago
- ☆26Feb 15, 2023Updated 3 years ago
- Experimental GPU language with meta-programming☆27Sep 6, 2024Updated last year
- An easy (and fast) API for popular 3D molecular datasets!☆46Feb 27, 2026Updated 3 weeks ago
- ☆21Mar 3, 2025Updated last year
- Pure C implementation of e3nn☆24Mar 17, 2025Updated last year
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 2025Updated last year
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Feb 25, 2025Updated last year
- PaiNN in jax☆11Jan 14, 2025Updated last year
- ☆25Mar 15, 2023Updated 3 years ago
- ☆32May 31, 2025Updated 9 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆89May 30, 2024Updated last year
- Benchmarking scripts for Gaia☆14Apr 10, 2025Updated 11 months ago
- ☆13Jun 2, 2024Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- A terminal-based citation generator☆14Jan 21, 2023Updated 3 years ago
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- An example project with Poetry and scikit-build.☆13Oct 29, 2024Updated last year
- ☆10Jul 16, 2020Updated 5 years ago
- Official repo of "Towards Interpretable Protein Structure Prediction with Sparse Autoencoders" published at ICLR 2025 GEM workshop.☆16Mar 13, 2025Updated last year
- ☆12Oct 10, 2023Updated 2 years ago
- Placeholder for the opensource Grid AI components☆45Jun 6, 2022Updated 3 years ago
- ☆47Nov 18, 2022Updated 3 years ago
- Integrate pytorch☆46Apr 4, 2021Updated 4 years ago
- Python bindings for OpenSHMEM☆25Mar 2, 2026Updated 3 weeks ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆16Nov 20, 2025Updated 4 months ago
- Tomasulo Simulator written in React as the project for Computer Architecture course, Spring 2019, Tsinghua University☆11Jun 9, 2019Updated 6 years ago
- Official codebase for "Context Aware Deep Learning for Multi Modal Depression Detection" [ICASSP 2019, Oral]☆11Dec 26, 2024Updated last year
- Pytorch Lightning seed project with hydra☆18Oct 8, 2020Updated 5 years ago
- Personal solutions to the Triton Puzzles☆20Jul 18, 2024Updated last year
- Display PackedPose objects, Pose objects, or PDB files within a Jupyter notebook and Google Colab☆11Oct 16, 2022Updated 3 years ago
- Lab materials for MIT 3.320 and Harvard AP 275 courses on atomistic modeling☆15Apr 20, 2021Updated 4 years ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆26Mar 11, 2026Updated last week
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 3 weeks ago
- Implementation of the SHA-3 family using AVX/AVX2 instructions.☆14Oct 5, 2018Updated 7 years ago
- Multi-Turn RL Training System with AgentTrainer for Language Model Game Reinforcement Learning☆60Dec 18, 2025Updated 3 months ago