Keras Python Packages

tqdm

:zap: A Fast, Extensible Progress Bar for Python and CLI

wandb

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

onnx

Open standard for machine learning interoperability

onnx2tf

A tool for converting ONNX files to LiteRT/TFLite/TensorFlow, PyTorch native code (nn.Module), TorchScript (.pt), state_dict (.pt), Exported Program (.pt2), and Dynamo ONNX. It also supports direct conversion from LiteRT to PyTorch.

torchinfo

View model summaries in PyTorch!

tensorflow-decision-forests

A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.

onnxmltools

ONNXMLTools enables conversion of models to ONNX

tf2onnx

Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX

scikeras

Scikit-Learn API wrapper for Keras.

onnx-weekly

Open standard for machine learning interoperability

pnnx

ncnn is a high-performance neural network inference framework optimized for the mobile platform

ncnn

ncnn is a high-performance neural network inference framework optimized for the mobile platform

keras-tuner

A Hyperparameter Tuning Library for Keras

neptune

📘 The experiment tracker for foundation model training

tdqm

:zap: A Fast, Extensible Progress Bar for Python and CLI

torchsummary

Model summary in PyTorch similar to `model.summary()` in Keras

tensorflow-model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

netron

Visualizer for neural network, deep learning and machine learning models

nudenet

Lightweight nudity detection

polyaxon

MLOps Tools For Managing & Orchestrating The Machine Learning LifeCycle

neptune-client

📘 The experiment tracker for foundation model training

horovod

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

opennsfw2

Keras implementation of the Yahoo Open-NSFW model

dagshub

DagsHub client libraries