View Alpaca_with_LoRA_bf16.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View Alpaca_with_LoRA.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View evaluate_LLM.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View rmm_mem_mgmt.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Initialize RMM pool on ALL workers | |
def _rmm_pool(): | |
rmm.reinitialize( | |
pool_allocator=True, | |
initial_pool_size=None, # Use default size | |
) | |
client.run(_rmm_pool) |
View nvtabular_gpu_cluster.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Deploy a Single-Machine Multi-GPU Cluster | |
protocol = "tcp" # "tcp" or "ucx" | |
visible_devices = "0,1,2,3" # Delect devices to place workers | |
device_spill_frac = 0.9 # Spill GPU-Worker memory to host at this limit. | |
# Reduce if spilling fails to prevent | |
# device memory errors. | |
cluster = None # (Optional) Specify existing scheduler port | |
if cluster is None: | |
cluster = LocalCUDACluster( |
View Train_a_Two_Tower_model_and_perform_offline_batch_inference.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View train_and_serve_an_XGBoost_model.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View cpu_vs_gpu_inference.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View XGBClassifier_add_noise.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View visualize_distribution.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
NewerOlder