Scalable Python DS & ML, in an API compatible & lightning fast way.
APACHE-2.0 License
Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021
A Python 3 library making time series data mining tasks, utilizing matrix profile algorithms, acc...
An Easy-to-use, Scalable and High-performance RLHF Framework (70B+ PPO Full Tuning & Iterative DP...
TensorLayerX: A Unified Deep Learning and Reinforcement Learning Framework for All Hardwares, Bac...
A research toolkit for particle swarm optimization in Python
A library of extension and helper modules for Python's data analysis and machine learning libraries.
[CAAI AIR'24] Bilateral Reference for High-Resolution Dichotomous Image Segmentation
BigDL: Distributed TensorFlow, Keras and PyTorch on Apache Spark/Flink & Ray
PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation
Sparse Boolean linear algebra for Nvidia Cuda, OpenCL and CPU computations
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed conf...
Stats, linear algebra and einops for xarray
DeepFloyd-IF (Imagen Free)
The Unified AI Framework