Stars
TensorDict is a pytorch dedicated tensor container.
Pan-cancer outcome prediction via a unified weakly supervised deep learning model
Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
Multimodal Prompting with Missing Modalities for Visual Recognition, CVPR'23
[ICCV2025] Official Pytorch Implementation of TinyViM
Compute CNN receptive field size in pytorch in one line
Python package built to ease deep learning on graph, on top of existing DL frameworks.
A node.js version management utility for Windows. Ironically written in Go.
mRMR (minimum-Redundancy-Maximum-Relevance) for automatic feature selection at scale.
Machine learning metrics for distributed, scalable PyTorch applications.
A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones
Compatibility layer for common array libraries to support the Array API
Pathology Foundation Model - Nature Medicine
Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning
Gpu accelerated vahadane stain normalization for Digital Pathology workflows.
Implementation of MusicLM, Google's new SOTA model for music generation using attention networks, in Pytorch
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
AshamaneCore Open Source MMO Framework (master = 9.0.2.36949, bfa = 8.3.7.35284, legion = 7.3.5.26972)
An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
Wallpaper engine PKG extractor/TEX to image converter
The fast version of DeLong's method for computing the covariance of unadjusted AUC.
Python/R library for feature selection in neural nets. ("Feature selection using Stochastic Gates", ICML 2020)
Code for the paper "Improved Techniques for Training GANs"