Stars
The Python Implementation of CRISP: Clustering Multi-Vector Representations for Denoising and Pruning
Benchmarking vision language vision on face tasks
Common CNN models defined for PyTorch Lightning
🛒 Simple recommender with matrix factorization, graph, and NLP. Beating the regular collaborative filtering baseline.
Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models
A hyperspherical face recognition library based on PyTorch
Segmentation models with pretrained backbones. PyTorch.
2nd place solution for the Generative Interior Design 2024 competition
A Python 3 script to harvest data from the Binance socket stream and calculate popular TA indicators and produce lists of top trending coins for use by algorithmic and bot traders storing data in a…
Rembg is a tool to remove images background
On the Unreasonable Effectiveness of Centroids in Image Retrieval
IPyPlot is a small python package offering fast and efficient plotting of images inside Python Notebooks. It's using IPython with HTML for faster, richer and more interactive way of displaying big …
A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
FinRL®: Financial Reinforcement Learning. 🔥
Some examples of using PyTorch for tabular data
Code for Machine Learning for Algorithmic Trading, 2nd edition.
Find your trading edge, using the fastest engine for backtesting, algorithmic trading, and research.
Official PyTorch Implementation of Proxy Anchor Loss for Deep Metric Learning, CVPR 2020
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
Streaming over lightweight data transformations
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
torch-optimizer -- collection of optimizers for Pytorch
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.