remote-gpu-trainer
Audited by Socket on Jul 24, 2026
3 alerts found:
Anomalyx3Overall, this is an orchestration wrapper for training runs with logging and artifact syncing. The most significant security concern is the intentional use of eval on PROXY_HOOK, which can become arbitrary command execution if PROXY_HOOK is not strictly controlled/trusted. The script also loads and exports a W&B API key from a local file, increasing the impact of any downstream leakage, but it does not itself perform network exfiltration or destructive actions beyond checkpoint housekeeping. No explicit malware payloads (backdoors/C2/crypto-mining) are evident in this fragment; risk is driven primarily by the eval trust boundary and the opaque behavior of the downstream python module executed with user-provided config/task inputs.
This module is primarily an orchestration/queue runner, not an overtly malicious payload in itself. However, it contains two high-impact execution primitives: (1) eval of a configurable PROXY_HOOK (defaulting to sourcing a local file), and (2) execution of a configurable runner script path via bash. If an attacker can influence PROXY_HOOK/DURABLE_DIR/RUN_ONE or the underlying sourced file contents, this becomes straightforward arbitrary code execution. Additionally, queue contents directly shape arguments passed into the runner without validation, increasing blast radius under queue tampering. Malware/exfiltration behaviors are not evident in this fragment, but supply-chain/ops compromise risk is significant due to dynamic execution.
This code is primarily a local reporting tool and does not show explicit malware behavior (no networking, persistence, or covert operations). However, it includes a clearly high-risk deserialization capability: when --allow-pickle is enabled, it can perform torch.load(..., weights_only=False) on checkpoint files from a user-supplied directory, which can lead to arbitrary code execution if best.pth is malicious. The security posture therefore depends heavily on treating the scanned checkpoint directory as fully trusted and ensuring --allow-pickle is never used on untrusted inputs.