Gaussian processes in TensorFlow
Single-cell analysis in Python
Probabilistic reasoning and statistical analysis in TensorFlow
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Message Passing Neural Networks for Molecule Property Prediction
Create videos with Stable Diffusion
Jittor is a high-performance deep learning framework
Making large AI models cheaper, faster and more accessible
DoWhy is a Python library for causal inference
Machine learning metrics for distributed, scalable PyTorch application
TFX is an end-to-end platform for deploying production ML pipelines
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
PyTorch extensions for fast R&D prototyping and Kaggle farming
Powering Amazon custom machine learning chips
Algorithmic Trading in Python with Machine Learning
Package of deep-learning based CTR models
Functional Machine Learning
A library of extension and helper modules for Python's data analysis
A Python Package to Tackle the Curse of Imbalanced Datasets in ML
JAX-based neural network library
Python Client for Supabase. Query Postgres from Flask, Django
Feature engineering package with sklearn like functionality
Python Optimal Transport
Lazy Predict help build a lot of basic models without much code
The machine learning toolkit for time series analysis in Python