Tensorflow Python Packages

datasets

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

onnxruntime

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

ray

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

wandb

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

thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

sagemaker

A library for training and deploying machine learning models on Amazon SageMaker

tensorflow

An Open Source Machine Learning Framework for Everyone

keras

Deep Learning for humans

onnx

Open standard for machine learning interoperability

sagemaker-core

A library for training and deploying machine learning models on Amazon SageMaker

etils

Collection of eclectic utils for python.

tensorflow-io-gcs-filesystem

Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO

onnxruntime-gpu

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

mtcnn

MTCNN face detection implementation for TensorFlow, as a PIP package.

mleap

MLeap: Deploy ML Pipelines to Production

cog

Containers for machine learning

coremltools

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

tensorflow-cpu

An Open Source Machine Learning Framework for Everyone

tensorflow-hub

A library for transfer learning by reusing parts of TensorFlow models.

open3d

Open3D: A Modern Library for 3D Data Processing

tensorflow-datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

sagemaker-train

A library for training and deploying machine learning models on Amazon SageMaker

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.