Geometric Computer Vision Library for Spatial AI

High-resolution models for human tasks.

Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers

[ECCV2024] Official implementation of paper, "DenseNets Reloaded: Paradigm Shift Beyond ResNets and ViTs".

Hackable and optimized Transformers building blocks, supporting a composable construction.

Segmentation models with pretrained backbones. PyTorch.

Personal framework for Deep Learning

Implement the paper "Self-Attention with Relative Position Representations"

[WACV 2024] Official PyTorch implementation of: Ordinal Classification with Distance Regularization for Robust Brain Age Prediction

Generate list of potential typo squatting domains with domain name permutation engine to feed AIL and other systems.

HMS - Harmful Brain Activity Classification

Kolmogorov Arnold Networks

torchsummaryX: Improved visualization tool of torchsummary

🦖Pytorch implementation of popular Attention Mechanisms, Vision Transformers, MLP-Like models and CNNs.🔥🔥🔥

Non-official implement of Paper：CBAM: Convolutional Block Attention Module

A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.

1st place solution to the Google - American Sign Language Fingerspelling Recognition competition

Accurately separates a URL’s subdomain, domain, and public suffix, using the Public Suffix List (PSL).

FastAPI Best Practices and Conventions we used at our startup

RETVec is an efficient, multilingual, and adversarially-robust text vectorizer.

Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning

Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams

Scrapy, a fast high-level web crawling & scraping framework for Python.

Convert Wikipedia database dumps into plaintext files

🤖 A PyTorch library of curated Transformer models and their composable components

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

🔥Highlighting the top ML papers every week.