Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
MIT License
Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and...
Implementation of Make-A-Video, new SOTA text to video generator from Meta AI, in Pytorch
A library for differentiable robotics.
One tensor operation is all you need
Implementation of SoundStorm, Efficient Parallel Audio Generation from Google Deepmind, in Pytorch
Unofficial implementation of iTransformer - SOTA Time Series Forecasting using Attention networks...
Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimiz...
Some personal experiments around routing tokens to different autoregressive attention, akin to mi...
KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling wit...
Explorations into the recently proposed Taylor Series Linear Attention
Implementation of Voicebox, new SOTA Text-to-speech network from MetaAI, in Pytorch
Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in Pytorch