94 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Vector (and Scalar) Quantization, in Pytorch | 1.7M | |
| A concise but complete full-attention transformer with a set of promising experi... | 1.4M | |
| Boltz | 63K | |
| Implementation of Alphafold 3 from Google Deepmind in Pytorch | 28K | |
| Implementation of Danijar's latest iteration for his Dreamer line of work | 19K | |
| Implementation of MeshGPT, SOTA Mesh generation using Attention, in Pytorch | 16K | |
| Implementation of rectified flow and some of its followup research / improvement... | 9K | |
| Implementation of π₀, the robotic foundation model architecture proposed by Phys... | 9K | |
| Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of th... | 8K | |
| Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch | 7K | |
| Implementation of the MetaController proposed in "Emergent temporal abstractions... | 7K | |
| Explorations into the proposed Streaming Deep Reinforcement Learning, from Unive... | 7K | |
| Fish Speech | 6K | |
| Implementation of HS-TasNet, "Real-time Low-latency Music Source Separation usin... | 6K | |
| Implementation of E(n)-Transformer, which incorporates attention mechanisms into... | 6K | |
| Protein design | 6K | |
| Shared utilities and training infrastructure for biomolecular structure predicti... | 6K | |
| Implementation of MultiScreen | 5K | |
| Implementation of F5-TTS in MLX | 5K | |
| Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Sho... | 5K | |
| Implementation of Soft Actor Critic and some of its improvements in Pytorch | 5K | |
| Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of ... | 4K | |
| LocoFormer - Generalist Locomotion via Long-Context Adaptation | 4K | |
| Implementation of a transformer for reinforcement learning using `x-transformers... | 4K | |
| The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few conve... | 4K | |
| Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Image... | 4K | |
| Boltz for VSYNTHES | 3K | |
| Recurrent Independent Mechanisms | 3K | |
| Native Sparse Attention | 3K | |
| Implementation and explorations into Blackbox Gradient Sensing (BGS), an evoluti... | 3K | |
| Implementation of Humanoid Standing Up, from the paper "Learning Humanoid Standi... | 3K | |
| Implementation of the Large Behavioral Model architecture for dexterous manipula... | 3K | |
| Implementation of the new SOTA for model based RL, from the paper "Improving Tra... | 3K | |
| Contrastive Reinforcement Learning | 3K | |
| H-Net Dynamic Chunking Modules | 3K | |
| Implementation of the training framework proposed in Self-Rewarding Language Mod... | 2K | |
| Memory-Augmented Sequence Models in Pytorch | 2K | |
| 2K | ||
| Pytorch implementation of the PEER block from the paper, Mixture of A Million Ex... | 2K | |
| Calculate, search, and analyze MIDI-to-MIDI similarity at scale | 2K | |
| Exploration into the proposed architecture from Sapient Intelligence of Singapor... | 2K | |
| Gaia2 - Pytorch | 2K | |
| Implementation of Mimic-Video, Video-Action Models for SOTA Generalizable Robot ... | 2K | |
| GotenNet in Pytorch | 2K | |
| Implementation of AlphaFold 3 in PyTorch Lightning + Hydra | 2K | |
| Implementation of Autoregressive Diffusion in Pytorch | 2K | |
| Equiformer - SE3/E3 Graph Attention Transformer for Molecules and Proteins | 1K | |
| Simple Replay Buffer for RL | 1K | |
| Score-based Diffusion models in JAX. | 1K | |
| TabTune: A Unified Library for Inference and Fine-Tuning Tabular Foundation Mode... | 1K |
