Clean Jupyter notebooks of outputs, metadata, and empty cells
Making DAG construction easier
Synthetic data generators for tabular and time-series data
AWS Lambda support for ASGI applications
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
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
Probabilistic programming in Python
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Linux port of FAR v2
Making Enterprise Data Intelligent and Responsive for AI
An AI agent development platform with all-in-one visual tools
High-performance library for gradient boosting on decision trees
Technical principles related to large models
ContextGem: Effortless LLM extraction from documents
Internet-scale Neural Networks
Benchmarking Multimodal Agents for Open-Ended Tasks
[NeurIPS 2023 Spotlight] LightZero
Advanced evolutionary computation library built on top of PyTorch
Implementation of RLHF (Reinforcement Learning with Human Feedback)
Build voice-based LLM agents. Modular + open source