Jupyter Lab widget around the https://github.com/fredvernier/DensityMaps project that uses webGPU to implement the Multiclass Density Map visualizations. See also https://github.com/e-/Multiclass-Density-Maps for the original (slow) implementation.
Clone the github project locally and install it with:
pip install -e .