Accelerated ML v1.0.0 LIVE
Google Cloud & NVIDIA Developer Codelab

Zero-Code-Change ML Acceleration with NVIDIA cuDF & cuML

Accelerating tabular machine learning on Google Cloud Colab Enterprise. Predict NYC yellow taxi tip amounts across millions of trips at 13.1x faster speed without rewriting standard pandas or scikit-learn code.

Google Cloud Colab Enterprise NVIDIA L4 Tensor Core GPU XGBoost CUDA Hist TLC NYC Taxi 2024
Overall Speedup
13.1x
24.0s (GPU) vs. 315.6s (CPU)
Ensemble RMSE
$0.9084
+$0.0131 lift over best model
Dataset Records
2.96M+
NYC Yellow Taxi Trip Records
Code Rewrites
0 lines
%load_ext cudf.pandas & cuml.accel

Pipeline Benchmark: CPU vs. GPU

Execution time in seconds for each phase on NVIDIA L4 vs. n2-standard CPU.

Dataset Scale Time-Savings Projection
Estimate time saved across different workload sizes:
CPU Duration
~63.1 mins
GPU Duration (cuDF/cuML)
~4.8 mins
Net Time Saved
⚡ 58.3 mins saved

Live NYC Taxi Tip Prediction Simulator

Simulate predictions from the three cross-validated regression models and the optimal weighted ensemble.

Fare Amount ($) $25.50
Trip Distance (miles) 4.8 miles
Predicted Tip Amount
$5.24
20.5% of Fare
XGBoost GPU (Weight: 52%) $5.31
cuML Random Forest (Weight: 36%) $5.18
cuML Linear Reg (Weight: 12%) $5.11

Code Execution Architecture

Explore each phase of the accelerated machine learning pipeline.

1 cuDF Ingestion & Cleaning

Parquet partitions read straight to GPU memory with column downcasting (float64->float32), slashing memory footprint by 50%.

%load_ext cudf.pandas
import pandas as pd
df = pd.read_parquet(...)
2 cuML Accelerated Models

Scikit-Learn algorithms automatically offloaded to NVIDIA GPUs for cross-validation without changing standard function signatures.

%load_ext cuml.accel
from sklearn.ensemble import ...
model.fit(X_train, y_train)
3 cuDF Profiling Magic

Line-by-line inspection identifies slow row-by-row CPU fallbacks (e.g. .apply()) to convert them into 100% vectorized GPU kernels.

%%cudf.pandas.line_profile
df['time'] = pd.cut(df['hour'], ...)
# 100% GPU vectorized

Ready to run on your own GPU environment?

Clone the repo, run in Google Colab Enterprise, or benchmark locally with RAPIDS.