Stars
A PyTorch native platform for training generative AI models
Causal depthwise conv1d in CUDA, with a PyTorch interface
A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
Resources about time series forecasting and deep learning.
Collection of tutorials to use the MACE machine learning force field.
Tools for building equivariant polynomials on reductive Lie groups.
Higher order equivariant graph neural networks for 3D point clouds
MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.