Library for metric learning pipelines and models.

A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos

so-vits-svc fork with realtime support, improved interface and more features.

A deep learning model to age faces in the wild, currently runs at 60+ fps on GPUs

High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial (Chebyshev nodes) and Fourier Series layers of arbitrary order. Piecewise implementations could be thought of as a 1d grid (for each neuron) where each grid element is Lagrange polynomial. Both full connected a…

A multitask neural network that does both regression and classification

Create powerful Hydra applications without the yaml files and boilerplate code.

A Toolbox in DOCSAID

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

PyTorch Lightning template for rapid, scalable and reproducible ML experimentation with best practices.

Online Handwritten Text Recognition (HTR) system implemented with PyTorch. Based on https://doi.org/10.1007/s10032-020-00350-4.

A Python Library for Uncertain Knowledge Graph Reasoning

From Python basics to Machine Learning and PyTorch Deep Learning - one day at a time, explore it all

(Arxiv 2024) Official implementation of Paper ''DreamView: Injecting View-specific Text Guidance into Text-to-3D Generation''

A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.

Open MatSci ML Toolkit is a framework for prototyping and scaling out deep learning models for materials discovery supporting widely used materials science datasets, and built on top of PyTorch Lightning, the Deep Graph Library, and PyTorch Geometric.

Config-based framework for organized and reproducible deep learning. MONAI Bundle + PyTorch Lightning.

A high-level library to make the training of Boltzmann Generators simple and reproducible.

Quadra: Effortless and reproducible deep learning workflows with configuration files.