arm-pytorch-utilities

Utilities for working with pytorch

arm-pytorch-utilities v0.5.0 123.1K downloads/30d#11,922 on PyPI★ 8

Permissive license Copyright (c) 2023 University of Michigan ARM Lab Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to… (full text in the JSON record) Active released 2026-03-10

What it is and what it does

arm-pytorch-utilities is a collection of helper functions for PyTorch development maintained by the University of Michigan ARM Lab. It provides utilities centered on random number generation control and visualization, designed to support reproducible machine learning experiments and aid in debugging PyTorch models.

The package depends on torch, numpy, matplotlib, scipy, and pytorch-seed, making it a natural fit for existing ML workflows. It requires Python 3.6+ and PyTorch 1.0+. Installation is straightforward via pip, and the project supports editable installs for development. The codebase is actively maintained with recent commits and has been in development since 2023-02-10.

Use it for:

Seed and control random number generation across PyTorch experiments to ensure reproducibility
Visualize model outputs, training dynamics, or intermediate representations using built-in plotting utilities
Standardize utility patterns across a team's PyTorch projects through a shared library
Debug neural network behavior by leveraging helper functions for common ML tasks

Worth the install?

AI-flagged interpretation of the facts on this page — verify before relying

Provides utility functions for PyTorch development, including random number generation control and visualization tools.

Yes, if you work regularly with PyTorch and need reproducible RNG control or visualization helpers. The package has low install friction, active maintenance, permissive licensing, and no known vulnerabilities. The small repository size suggests it is specialized rather than general-purpose, so evaluate whether its specific utilities match your workflow.

Install

arm-pytorch-utilities on PyPI

pip

pip install arm-pytorch-utilities

uv

uv add arm-pytorch-utilities

poetry

poetry add arm-pytorch-utilities

Installing arm-pytorch-utilities

Before you install

Low friction installation with a pure Python wheel. Actively maintained with recent commits as of 2026-03-10. Five runtime dependencies (torch, numpy, matplotlib, scipy, pytorch-seed) are standard in ML environments.

License in practice

MIT license permits unrestricted use, modification, and distribution with only attribution and liability disclaimer required—suitable for both open and commercial projects.

Quickstart

pip install arm-pytorch-utilities

import arm_pytorch_utilities
import torch

Requires PyTorch 1.0+ and Python 3.6+

Verify before relying

Specific utility functions and their use cases beyond rng and visualization
Whether pytorch-seed is a required runtime dependency or optional
API stability and backward compatibility guarantees across versions

Package facts

License	Copyright (c) 2023 University of Michigan ARM Lab Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to… (full text in the JSON record) (permissive)
Python support	supports the current Python release (>=3.6)
Install friction	low — pure-Python wheel
Runtime dependencies	5 — torch, numpy, matplotlib, scipy, pytorch-seed
Maintenance	actively maintained — 157 days since the last release
Last repo commit	2026-03-10
First released	2023-02-10
Downloads	123,137/month — #11,922 on PyPI (30-day window, as of 2026-08-14)
Known vulnerabilities	none known (OSV.dev, checked 2026-08-14)

Evidence: arm_pytorch_utilities-0.5.0-py3-none-any.whl

Keywords: rng, pytorch, visualization

Development Status :: 4 - BetaIntended Audience :: DevelopersLicense :: OSI Approved :: MIT LicenseProgramming Language :: Python :: 3Programming Language :: Python :: 3 :: Only