Repository of the paper "AnyUp: Universal Feature Upsampling".

An extremely fast Python package and project manager, written in Rust.

[CVPR 2024] GSNeRF: Generalizable Semantic Neural Radiance Fields with Enhanced 3D Scene Understanding

Cosmos-Predict2 is a collection of general-purpose world foundation models for Physical AI that can be fine-tuned into customized world models for downstream applications.

Slide decks, coding exercises, and quick references for learning the JAX AI Stack

Dino V2 for Classification, PCA Visualization, Instance Retrival: https://arxiv.org/abs/2304.07193

Library for reading and processing ML training data.

a Jax quantization library

Pretrained Keras 3 vision models with custom layers, including object detection, segmentation, and vision-language models (VLMs), with backbone support.

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, successor to TensorFlow Lite. is Google's On-device framework for high-performance ML & GenAI deployment on edge platforms, via efficient conversion, runtime, and optimization

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.