A library for accelerating Transformer models on NVIDIA GPUs
A comprehensive set of fairness metrics for datasets
Data loaders and abstractions for text and NLP
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
High-performance library for gradient boosting on decision trees
Lightning fast C++/CUDA neural network framework
Solve end to end problems using Llama model family
A TensorFlow library for applied reinforcement learning
Multilingual Automatic Speech Recognition with word-level timestamps
Superfast AI decision making and processing of multi-modal data
A system for quickly generating training data with weak supervision
Open source platform for the machine learning lifecycle
Modular quant framework
AI discovers 520000 stable inorganic crystal structures for research
Training data (data labeling, annotation, workflow) for all data types
Spatiotemporal Signal Processing with Neural Machine Learning Models
Data science on data without acquiring a copy
Python package built to ease deep learning on graph
Standalone, small, language-neutral
Reference implementations of MLPerf™ training benchmarks
TFX is an end-to-end platform for deploying production ML pipelines
TimeGPT-1: production ready pre-trained Time Series Foundation Model