A standard API for reinforcement learning and a diverse set of reference environments (formerly Gym).
12 similar packages · ranked by shared classifiers & health score · View gymnasium →
A CLI and library for interacting with the Weights & Biases API.
A hyperparameter optimization framework
Gymnasium for multi-agent reinforcement learning.
NVIDIA cuDNN Frontend — Python and C++ Graph API with SOTA attention (SDPA / Flash Attention), MoE grouped GEMM fusions, and FP8/MXFP8 kernels for Hopper and Bl
Integration libraries of Optuna.
spaCy pipelines for pre-trained BERT and other transformers
A library for efficient similarity search and clustering of dense vectors.
Bayesian networks and other Probabilistic Graphical Models.
TF.Text is a TensorFlow library of text related ops, modules, and subgraphs.
A library for efficient similarity search and clustering of dense vectors.
Fast inference engine for Transformer models
The Arcade Learning Environment (ALE) - a platform for AI research.