Deep Learning Library Python Packages

nvidia-cutlass-dsl-libs-base

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl-libs-cu13

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl-libs-cu12

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl-libs-core

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-operators

CUDA Templates and Python DSLs for High-Performance Linear Algebra

cerium-delta

Real-time neural network observability platform for monitoring activations, gradients, information flow, and architecture health.

dandelion

A light weight deep learning framework, on top of Theano, offering better balance between flexibility and abstraction

lasagne

Lightweight library to build and train neural networks in Theano

sparseml

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

torch-geopooling

The geospatial pooling modules for neural networks in PyTorch

nutsml

Flow-based data pre-processing for deep learning

starttf

A tensorflow batteries included kit to write tensorflow networks from scratch or use existing ones.

nets

Recreating PyTorch from scratch, using Numpy. Supports FCN, CNN, RNN layers.

ntk-ml

A from-scratch neural network and tensor library built in Python for learning, experimentation, and deep understanding of modern ML systems.

shinnosuke

A keras-like API deep learning framework,realized by Numpy only.Support CNN, RNN, LSTM, Dense, etc.

kd-lib

A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.

nvidia-cutlass-dsl-libs-qnx-cu13

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl-libs-qnx-cu12

CUDA Templates and Python DSLs for High-Performance Linear Algebra

n3d3

N2D2 is an open source CAD framework for Deep Neural Network simulation and full DNN-based applications building.

tinynn

A lightweight deep learning library

whiteboxlayer

Deep Learning Package based on TensorFlow.

torchhandle

TorchHandle makes your PyTorch development more efficient and make you use PyTorch more comfortable