72 dependents
Package Description Downloads/month
PyTorch native post-training library 410K
dmlc dgl
Python package built to ease deep learning on graph, on top of existing DL frame... 137K
verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework 123K
A PyTorch native platform for training generative AI models 55K
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024) 29K
MatGL is a framework for graph deep learning for materials science. 28K
A Ray-based High-performance RLHF framework. 10K
Lean LLM fine-tuning toolkit with EC2 orchestration for SFT, DPO, and LoRA train... 4K
🚀 Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Fa... 3K
d9d-project d9d
d9d - d[istribute]d - distributed training framework based on PyTorch that tries... 3K
Toolkit for structure-based deep learning on RNA. 3K
Self-contained distributed community captioning system 3K
a dataloader to work with large single cell datasets from lamindb 3K
Multimodal datasets for spatial intelligence 2K
2K
VeOmni: Scaling Any Modality Model Training with Model-Centric Distributed Recip... 2K
HSCI research group utilities for pytorch (lightning) 2K
A unified inference and post-training framework for accelerated video generation... 2K
Contrastive pre-training for technology-agnostic single-cell representations bey... 1K
single cell foundation model for Gene network inference and more 1K
Adapter to bridge DBInfer datasets and tasks to the RelBench interface with cach... 946
基于 OpenKE-PyTorch 开发的知识图谱嵌入工具包 810
a simpler version of flashattention 785
Semantic segmentation for plant roots grown in EcoFABs 770
GraphStorm 736
Modular, high-performance deep learning research framework 699
KAPipe: A modular pipeline for knowledge acquisition. 649
Flow Gym 624
Automated machine-learned Potential Landscape explorer 594
BackdoorMBTI is an open source project expanding the unimodal backdoor learning ... 567
scPRINT-2 is a Large Cell Model for Gene Network Inference, Denoising and more f... 554
modlee package 476
https://atomgpt.org 474
Our version of the NIST-JARVIS ALIGNN optimized in terms of model performance an... 457
tinybig for deep function learning 443
Interpretable Analysis of Histological WSIs 441
minimal deep learning framework 424
390
OneScience是基于先进的深度学习框架打造的科学计算工具包,旨在通过一系列高度集成的组件加速科学研究和技术开发进程。 388
A Python module for generating Bangla IPA transliterations. 385
Experimaestro module for Pytorch-based experiments 381
neural force field learning toolkit 355
siiRL: Shanghai Innovation Institute RL Framework for Advanced LLMs and Multi-Ag... 348
XLM Framework 335
A Python package for Graph Intellectual Property Protection. 324
The 🌏 data science library you've been waiting for~ 312
skyrl-train 285
Unit HiFi-GAN 252
A PyTorch native platform for post-training generative AI models 246
Reinforcement Learning Infrastructure for Embodied and Agentic AI 245