PyPI page
Home page
Author:
None
License:
MIT License
Copyright (c) 2026 ChharithOeun
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated docum...
Summary:
Universal GPU setup and diagnostics for PyTorch and JAX — DirectML, ROCm, CUDA, MPS, CPU. Auto-detects your hardware and sets HSA_OVERRIDE_GFX_VERSION for unsupported AMD GPUs.
Latest version:
1.0.0
Optional dependencies:
flax
|
jax
|
optax
|
torch
Downloads last day:
113
Downloads last week:
127
Downloads last month:
159