Deeplearning Python Packages

nemo-toolkit

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

clearml

ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution

clearml-agent

ClearML Agent - MLOps/LLMOps made easy. MLOps/LLMOps scheduler & orchestration solution

voxcpm

VoxCPM2: Tokenizer-Free TTS for Multilingual Speech Generation, Creative Voice Design, and True-to-Life Cloning

netron

Visualizer for neural network, deep learning and machine learning models

pytabkit

ML models + benchmark for tabular data classification and regression

ptflops

Flops counter for neural networks in pytorch framework

urduhack

An NLP library for the Urdu language. It comes with a lot of battery included features to help you process Urdu data in the easiest way possible.

horovod

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

qai-hub-models

Qualcomm® AI Hub Models is our collection of state-of-the-art machine learning models optimized for performance (latency, memory etc.) and ready to deploy on Qualcomm® devices.

albumentationsx

Next-generation Albumentations: dual-licensed for open-source and commercial use

pycm

Multi-class confusion matrix library in Python

qai-hub-models-cli

Qualcomm® AI Hub Models is our collection of state-of-the-art machine learning models optimized for performance (latency, memory etc.) and ready to deploy on Qualcomm® devices.

ludwig

Low-code framework for building custom LLMs, neural networks, and other AI models

polyaxon-schemas

Schema definitions and validation for Polyaxon.

vectorhub-nightly

Vector Hub - Library for easy discovery, and consumption of State-of-the-art models to turn data into vectors. (text2vec, image2vec, video2vec, graph2vec, bert, inception, etc)

dicomrttool

Tools to help with the conversion of DICOM images, RT Structures, and dose to useful Python objects. Essentially DICOM to NumPy and SimpleITK Images

torchensemble

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.

retinaface-pytorch

The remake of the https://github.com/biubug6/Pytorch_Retinaface

bert-classifier

A repository for training transformer based models

airsim

Open source simulator for autonomous vehicles built on Unreal Engine / Unity, from Microsoft AI & Research

keras-unet

Helper package with multiple U-Net implementations in Keras as well as useful utility tools helpful when working with image semantic segmentation tasks. This library and underlying tools come from multiple projects I performed working on semantic segmentation tasks

nlp-models

A repository for training transformer based models

auto-ml

[UNMAINTAINED] Automated machine learning for analytics & production