Search Packages

ultralytics-thop

Profile PyTorch models for FLOPs and parameters, helping to evaluate computational efficiency and memory usage.

easyocr

Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

omlt

Represent trained machine learning models as Pyomo optimization formulations

tsai

Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai | State-of-the-art Deep Learning library for Time Series and Sequences in Pytorch / fastai

torchcam

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

attacut

A Fast and Accurate Neural Thai Word Segmenter

torchnd

ND convolutions, complex convolutions, ND padding, etc

melpy

A NumPy-based deep learning library for building neural networks. It features an automatic differentiation engine and supports training architectures like LSTM, CNN, and FNN.

neuralnetlib

A flexible deep learning framework built from scratch using only NumPy

keras-vision

Porting vision models to Keras 3 for easily accessibility. Contains MobileViT v1, MobileViT v2, fastvit

continual-inference

A Python library for Continual Inference Networks in PyTorch

yolox-backbone

yolox_backbone is a deep-learning library and is a collection of YOLOX Backbone models.

topaz-em

Particle picking with positive-unlabeled CNNs

moai-mdk

moai: Accelerated, Flexible, Modular, Reproducible, Insightful AI

basic-image-eda

A simple image dataset EDA tool (CLI / Code)

diploshic

A deep learning tool for identifying hard and soft selective sweeps in population genomic data

gcvit

Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer

jai

Just Assemble IT! - A LEGO-style & PyTorch-based Deep Learning Library

efficientword-net

OneShot Learning-based hotword detection.

tsfedl

Time Series Feature Extraction using Deep Learning

cnn4ie

Chinese Information Extraction

planer

Powerful Light Artificial NEuRon inference framework for CNN

tempocnn

Framework for estimating temporal properties of music tracks.

torch-yolo3

YOLO v3 in PyTorch