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.

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

einops

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

tensorflow

An Open Source Machine Learning Framework for Everyone

sagemaker

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

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

mleap

MLeap: Deploy ML Pipelines to Production

cog

Containers for machine learning

tensorflow-hub

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

coremltools

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

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.

tensorflow-cpu

An Open Source Machine Learning Framework for Everyone

supervision

We write your reusable computer vision tools. 💜