Ongoing research training transformer models at scale

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

CMSIS-DSP embedded compute library for Cortex-M and Cortex-A

LiteRT is the new name for TensorFlow Lite (TFLite). While the name is new, it's still the same trusted, high-performance runtime for on-device AI, now with an expanded vision.

Docker implementation of TensorFlow and EfficientNet v2 for training wildlife camera trap classifiers

MEWC: Mega Efficient Wildlife Classifier

Official code for Metric learning for user-defined keyword spotting

CMSIS-NN Library

A Keras 3 Implementation of AdEMAMix Optimizer

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Pretrained model hub for Keras 3.

🚴 Call stack profiler for Python. Shows you why your code is slow!

An open-source, easily accessible package for training and deploying Speech-to-Intent models on microcontrollers and SBCs

The fundamental package for scientific computing with Python.

Implementation of "Recurrent Vision Transformers for Object Detection with Event Cameras". CVPR 2023

The official implementation of the ICRA 2024 paper "V2CE: Video to Continuous Events Simulator." This repo provides a out-of-the-box tool that you can easily and swiftly convert your video to highl…

OpTree: Optimized PyTree Utilities

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extraction, and more.

TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.

Deep Learning for humans

⚡ A Fast, Extensible Progress Bar for Python and CLI

A demonstration of using a live Tensorflow session to create an interactive face-GAN explorer.

Visualizer for neural network, deep learning and machine learning models

A library that includes Keras 3 preprocessing and augmentation layers, providing support for various data types such as images, labels, bounding boxes, segmentation masks, and more.

Parse TFLite models (*.tflite) EASILY with Python. Check the API at https://zhenhuaw.me/tflite/docs/

Probabilistic reasoning and statistical analysis in TensorFlow