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

onnx-weekly

Open standard for machine learning interoperability

tf2onnx

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

scikeras

Scikit-Learn API wrapper for Keras.

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

tdqm

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

neptune

📘 The experiment tracker for foundation model training

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

keras-ocr

A packaged and flexible version of the CRAFT text detector and Keras CRNN recognition model.

torch-summary

View model summaries in PyTorch!

opennsfw2

Keras implementation of the Yahoo Open-NSFW model

horovod

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

dagshub

DagsHub client libraries