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 →
Gymnasium for multi-agent reinforcement learning.
A CLI and library for interacting with the Weights & Biases API.
NVIDIA cuDNN Frontend — Python and C++ Graph API with SOTA attention (SDPA / Flash Attention), MoE grouped GEMM fusions, and FP8/MXFP8 kernels for Hopper, Black
A hyperparameter optimization framework
Integration libraries of Optuna.
A library for efficient similarity search and clustering of dense vectors.
spaCy pipelines for pre-trained BERT and other transformers
Bayesian networks and other Probabilistic Graphical Models.
Parameter-Efficient Fine-Tuning (PEFT)
Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and
High-performance image processing functions for deep learning and computer vision.
TensorFlow is an open source machine learning framework for everyone.