7 projects
envpool
"C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments."
grain-free-threading
Grain: A library for loading and transforming data for ML training.
pylance-free-threading
python wrapper for Lance columnar format
ami-torchcodec
A video decoder for PyTorch
ami-decord
Decord Video Loader
autofd
"Automatic Functional Derivative in JAX"