Neural Networks Python Packages

onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

spacy

💫 Industrial-strength Natural Language Processing (NLP) in Python

blis

💥 Fast matrix-multiplication as a self-contained Python library – no system dependencies!

keras

Deep Learning for humans

treescope

An interactive HTML pretty-printer for machine learning research in IPython notebooks.

onnxruntime-gpu

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

tensorflow-probability

Probabilistic reasoning and statistical analysis in TensorFlow

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)

keras-hub

Deep Learning for humans

onnxruntime-openvino

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

tfp-nightly

Probabilistic reasoning and statistical analysis in TensorFlow

torch-ema

Tiny PyTorch library for maintaining a moving average of a collection of parameters.

optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

dm-control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

gluonts

Probabilistic time series modeling in Python

keras-nightly

Deep Learning for humans

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

dm-haiku

JAX-based neural network library

adaptive-classifier

A flexible, adaptive classification system for dynamic text classification

onnxruntime-directml

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

rtdl-num-embeddings

(NeurIPS 2022) On Embeddings for Numerical Features in Tabular Deep Learning

keras-nlp

Deep Learning for humans

chemprop

Message Passing Neural Networks for Molecule Property Prediction