xfcui
Follow
Stars
⚡FlashRAG: A Python Toolkit for Efficient RAG Research (WWW2025 Resource)
End-To-End Molecular Dynamics (MD) Engine using PyTorch
Differentiable, Hardware Accelerated, Molecular Dynamics
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
A modular framework for neural networks with Euclidean symmetry
A powerful and flexible machine learning platform for drug discovery
Fast and memory-efficient exact attention
Source code for From Stars to Subgraphs (ICLR 2022)
Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, Materials Science and Biology
Graph Neural Network Library for PyTorch
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Optax is a gradient processing and optimization library for JAX.
Flax is a neural network library for JAX that is designed for flexibility.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A library for efficient similarity search and clustering of dense vectors.