Natural Gradient Boosting for Probabilistic Prediction
Gaussian processes in TensorFlow
Single-cell analysis in Python
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
BitNet: Scaling 1-bit Transformers for Large Language Models
Jupyter notebook tutorials for OpenVINO
Scalable and user friendly neural forecasting algorithms.
Making large AI models cheaper, faster and more accessible
The easiest way to use deep metric learning in your application
Geometric deep learning extension library for PyTorch
A general fine-tuning kit geared toward image/video/audio diffusion
Hummingbird compiles trained ML models into tensor computation
An open source python library for automated feature engineering
This repos contains notebooks for the Advanced Solutions Lab
A unified interface for distributed computing
Algorithms for explaining machine learning models
A Hyperparameter Tuning Library for Keras
TFDS is a collection of datasets ready to use with TensorFlow,
Multi-class confusion matrix library in Python
Build portable, production-ready MLOps pipelines
Open deep learning compiler stack for cpu, gpu, etc.
A library for scientific machine learning & physics-informed learning
Tool for visualizing and tracking your machine learning experiments
A game theoretic approach to explain the output of ml models
A Python library for learning and evaluating knowledge graph embedding