73 dependents
Package Description Downloads/month
Pytorch domain library for recommendation systems 1.4M
A modular, primitive-first, python-first PyTorch library for Reinforcement Learn... 1.3M
cortical is the framework for building fabric architectures 325K
A fast and simple implementation of learning algorithms for robotics. 131K
verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework 122K
Open-source deep-learning framework for building, training, and fine-tuning deep... 40K
Isaac Lab API, powered by MuJoCo-Warp, for RL and robotics research 20K
AgileRL is a deep reinforcement learning library focused on improving RL develop... 14K
Sequence correction provided by ONT Research 11K
Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch 7K
Neural network sequence error correction. 5K
Infrastructure simulation library for reinforcement learning and optimization 5K
A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial O... 4K
Generative Quantum Circuits 4K
Trinity-RFT is a general-purpose, flexible and scalable framework designed for r... 3K
Memory-Augmented Sequence Models in Pytorch 2K
2K
AI on the way 2K
TransferQueue: An Asynchronous Streaming Data Management Module 2K
Multimodal datasets for spatial intelligence 2K
Cosmos-RL is a flexible and scalable Reinforcement Learning framework specialize... 2K
A torch implementation of GFlowNets 2K
An end-to-end, GPU-accelerated, and modular platform for building generalized Em... 2K
🔁 AMP-RSL-RL: Adversarial Motion Priors for robotic RL (PPO + motion imitation) 1K
High-performance open-source deep-learning framework for exploring, building and... 1K
Deep Multiple Instance Learning library for Pytorch 1K
Library for quantum cognition machine learning. 1K
Reinforcement Learning toolkit for Assetto Corsa and Car Racing environments 1K
JSON -> [*] 1K
Interpretability for LeelaChessZero networks. 1K
Library for quantum cognition machine learning. 993
Various scene understanding and perception evaluation metrics. 952
Uncertainty quantification with PyTorch 921
Implementation of a random selection of artificial neural net based models. 869
This library provides tools and utilities for Few Shot Learning in Natural Langu... 864
Iterable Streaming Webdataset for PyTorch from boto3 compliant storage 863
Tools and libraries for unified multi-task vision research. 790
A modular robot training frameworks for Genesis inspired by Isaac Lab and Gymnas... 779
A framework built on PyTorch for eco-hydrological modeling 763
cortical is a modular library for building recurrent backbones and agent memory ... 728
657
A Taichi Gaussian Splatting library 637
🤖🪝Interpretability with tensordict and torch hooks. 598
A multi-tenant fine-tuning platform for LLMs with Tinker-compatible API 593
Online cache for prompt embeddings using TensorDict memory-mapped storage 539
MuJoCo MJX ported to PyTorch 532
A generative AI framework for 3D molecular generation and data-driven molecular ... 510
Cooldata - A Large-Scale Electronics Cooling 3D Flow Field Dataset 481
A viewer for gaussian-splatting reconstructions 402
siiRL: Shanghai Innovation Institute RL Framework for Advanced LLMs and Multi-Ag... 348