Search
xgboost
Hyunsu Cho <[email protected]>, Jiaming Yuan <[email protected]>
XGBoost Python Package
lightgbm
LightGBM Python-package
wandb
Weights & Biases <[email protected]>
A CLI and library for interacting with the Weights & Biases API.
optuna
Takuya Akiba
A hyperparameter optimization framework
ray
Ray Team
Ray provides a simple, universal API for building distributed applications.
pytorch-lightning
Lightning AI et al.
PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
nltk
NLTK Team
Natural Language Toolkit
gensim
Radim Rehurek
Python framework for fast Vector Space Modelling
flair
Alan Akbik
A very simple framework for state-of-the-art NLP
textblob
Steven Loria <[email protected]>
Simple, Pythonic text processing. Sentiment analysis, part-of-speech tagging, noun phrase parsing, and more.
pattern
Sane Punk
pattern is a Python package offering a suite of plotting functions to visualize machine learning models and data. It provides intuitive and customizable plots to aid in model evaluation and data analysis.
...moreallennlp
Allen Institute for Artificial Intelligence
An open-source NLP research library, built on PyTorch.
simpletransformers
Thilina Rajapakse
An easy-to-use wrapper library for the Transformers library.
ChatterBot
Gunther Cox
ChatterBot is a machine learning, conversational dialog engine
llama-cpp-python
Andrei Betlen <[email protected]>
Python bindings for the llama.cpp library
ctransformers
Ravindra Marella
Python bindings for the Transformer models implemented in C/C++ using GGML library.
exllamav2
turboderp
No description available
auto-gptq
PanQiWei
An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.
bitsandbytes
Tim Dettmers <[email protected]>
k-bit optimizers and matrix multiplication routines.
airflow
Apache Author
Placeholder for the old Airflow package