SUMMARY sorix _version autograd backend clustering k_means cuda cuda cupy cupy metrics metrics model_selection train_test nn init layers loss net optim lr_scheduler optim scipy_bridge preprocessing encoders scalers transformers tensor utils data dataloader dataset math profiling utils