64 dependents
Package Description Downloads/month
Audiocraft is a library for audio processing and generation with deep learning. ... 122K
SchNetPack - Deep Neural Networks for Atomistic Systems 35K
MedVAE is a family of six medical image autoencoders that can encode high-dimens... 26K
HyFI: Hydra Fast Interface (Hydra and Pydantic based interface framework) 6K
🍵 Matcha-TTS: A fast TTS architecture with conditional flow matching 5K
A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial O... 5K
4K
A framework for exploring the depths of hyperparameter space with Hydra and MLfl... 3K
Retrieve, Read and LinK: Fast and Accurate Entity Linking and Relation Extractio... 2K
Collection of representation learning models, techniques, callbacks, utils, used... 2K
Implementation of AlphaFold 3 in PyTorch Lightning + Hydra 2K
A simple and efficient Minecraft Agent development kit for AI research. 2K
123D: A Unified Library for Multi-Modal Autonomous Driving Data 2K
An auto-ML python library for computer vision based on pytorch lightning and hyd... 2K
Quadra: Effortless and reproducible deep learning workflows with configuration f... 1K
🤖🌊 aiFlows: The building blocks of your collaborative AI 1K
Python package to create physics-based pedestrian models from crowd measurements 1K
CpGPT: a Foundation Model for DNA Methylation. 933
Differential analysis of targeted Isotope-labeled Metabolomics data 835
自然语言处理(NLP) 821
A MEDS PyTorch Dataset, leveraging a on-the-fly retrieval strategy for flexible,... 682
Comprehensive benchmarking of protein-ligand structure prediction methods 664
Framework for training MCMC samplers for Lattice QCD 569
Hydra sweeper integration of our favorite optimization packages, utilizing ask-a... 565
Code for `LLM2VEC-GEN: Generative Embeddings from Large Language Models` 532
A generative AI framework for 3D molecular generation and data-driven molecular ... 528
Code for the paper Hybrid Spectrogram and Waveform Source Separation 452
minimal deep learning framework 424
Benchmarking framework for protein representation learning. Includes a large num... 371
Marine: Multi-task learning based on Japanese accent estimation (Also supports W... 348
Golden Retriever - A Lightning framework for retriever architecture prototype 342
XLM Framework 335
Benchmark of Generative Large Language Models in Danish 325
A MEDS PyTorch Dataset, leveraging a on-the-fly retrieval strategy for flexible,... 311
Client and Protocols for DARPA sail-on 308
ZEBRA: Zero-Shot Example-Based Retrieval Augmentation for Commonsense Question A... 277
Medical Meta Learner - Leveraging knowledge from previous tasks 255
BirdSet: A multi-task benchmark and data pipeline for deep learning based avian ... 207
A modular, meta-learning-ready RL library. 206
A modular toolbox for meta-learning research with a focus on speed and reproduci... 204
Python Boilerplate that contains all the code you need to create a Python packag... 203
Python package for the validation of AROS [semi]structured data using pydantic m... 194
Nvbenjo is a utility for benchmarking inference of deep learning models on NVIDI... 193
Machine learning made easy 180
Explain Any Image Classifier with Trained Perturbations 180
Use NNC compute resource from NNabla 170
A Framework for Comparing N Hyperparameter Optimizers on M Benchmarks. 154
Audiocraft is a library for audio processing and generation with deep learning. ... 150
An open-source machine learning package for continual learning research, which h... 146
Library for machine learning on physical tensors 142