Model Optimization Python Packages

onnx-ir

Efficient in-memory representation for ONNX, in Python

sklearn-genetic-opt

Hyperparameter optimization and feature selection for scikit-learn using evolutionary algorithms. A modern alternative to GridSearchCV and RandomizedSearchCV.

floppy-tracker

A hardware-agnostic profiler for tracking the FLOPs and BOPs of Machine and Deep Learning algorithms.

bayeshalvingsearchcv

scikit-learn-compatible hyperparameter search: BayesHalvingSearchCV (or HalvingBayesianSearchCV - Bayesian/GP) and PatternSearchCV (Hooke-Jeeves), with multi-fidelity data growth for faster tuning on large datasets

sk-autod

Auto-diagnose deep learning training pathologies: overfitting, unstable LR, exploding gradients, noisy training, and early stopping issues.

mlflow-assist

A comprehensive toolkit for ML and LLM development