Tensorflow Python Packages

onnxruntime

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

datasets

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

einops

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

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

wandb

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

ray

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

onnx

Open standard for machine learning interoperability

tensorflow

An Open Source Machine Learning Framework for Everyone

keras

Deep Learning for humans

etils

Collection of eclectic utils for python.

sagemaker-core

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

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

open3d

Open3D: A Modern Library for 3D Data Processing

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.

supervision

We write your reusable computer vision tools. 💜

tensorflow-probability

Probabilistic reasoning and statistical analysis in TensorFlow

sagemaker-train

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

tensorflow-datasets

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