Optax is a gradient processing and optimization library for JAX
This repos contains notebooks for the Advanced Solutions Lab
Images to inference with no labeling
TFX is an end-to-end platform for deploying production ML pipelines
Pretrained (Language) Models for Probabilistic Time Series Forecasting
A high performance implementation of HDBSCAN clustering
JAX-based neural network library
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
A very simple framework for state-of-the-art NLP
Investment Research for Everyone, Everywhere
Build MLOps Pipelines in Minutes
PyTorch extensions for fast R&D prototyping and Kaggle farming
Neural machine translation and sequence learning using TensorFlow
OpenMMLab Image Classification Toolbox and Benchmark
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
Spatiotemporal Signal Processing with Neural Machine Learning Models
ONNX-TensorRT: TensorRT backend for ONNX
Probabilistic reasoning and statistical analysis in TensorFlow
Embed images and sentences into fixed-length vectors
Interactive deep learning book with multi-framework code
Geometric deep learning extension library for PyTorch
The unified and scalable ML library for large-scale training
A system for quickly generating training data with weak supervision
PyTorch version of Stable Baselines