AIMET is a library that provides advanced quantization and compression
A library for accelerating Transformer models on NVIDIA GPUs
Build portable, production-ready MLOps pipelines
Multilingual Automatic Speech Recognition with word-level timestamps
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Open source framework for deep learning satellite and aerial imagery
Images to inference with no labeling
Build MLOps Pipelines in Minutes
Deep and Machine Learning for Microscopy
ONNX-TensorRT: TensorRT backend for ONNX
Optax is a gradient processing and optimization library for JAX
An easy-to-use & supercharged open-source experiment tracker
A simple but complete full-attention transformer
Lightning fast C++/CUDA neural network framework
A game theoretic approach to explain the output of ml models
Solve end to end problems using Llama model family
Scalable and user friendly neural forecasting algorithms.
Your open-source LLM evaluation toolkit
The easiest way to use deep metric learning in your application
Data loaders and abstractions for text and NLP
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
Standalone, small, language-neutral
Reference implementations of MLPerf™ training benchmarks
TFX is an end-to-end platform for deploying production ML pipelines