SHAP v0.54.0

Highlights

CUDA acceleration with shap-cuda

GPU support now lives in the separate shap-cuda package, providing the CUDA-accelerated backend for shap.GPUTreeExplainer. The GPU wheel workflow builds CUDA-enabled wheels and tests them on AWS GPU runners before publishing. By @CloseChoice in #5140.

The GPU TreeExplainer notebook has also been refreshed by @RAMitchell in #5220.

What's Changed

Enhancements

• shap.plots.violin now accepts an ax argument and returns the axes, making it easier to customize plots and compose subplots. By @adt-kmr in #5218.

Maintenance

• Remove unused global-path-dependent Tree SHAP code. By @RAMitchell in #5223.

Documentation

• Fix typos in the "An introduction to explainable AI" notebook. By @Yaqiao-chi-helen in #5225.
• Fix typos in comments, docstrings, and an error message. By @guptaishaan in #5213.
• Fix typos and replace the deprecated API reference in the LinearExplainer correlation notebook with maskers.Impute(data). By @ojuschugh1 in #5172.

New Contributors

• @Yaqiao-chi-helen made their first contribution in #5225.
• @guptaishaan made their first contribution in #5213.
• @ojuschugh1 made their first contribution in #5172.

Full Changelog: v0.53.0...v0.54.0