PulseML

Pulse — a live ML training debugger, GUI or CLI, any backend.

Pulse is a live machine learning training debugger designed to monitor tensors, track metrics, visualize heatmaps and line charts, and interact with an integrated AI analyst.

Key Features

GUI Mode — Opens an interactive matrix picker with live shapes, followed by a live dashboard with a heatmap grid and an integrated AI chat panel.
Smart Scalars — Scalars (loss, accuracy, learning rate) automatically render as live step-charts rather than heatmaps. Loss-like scalars are auto-detected and pre-selected in the picker.
CLI Mode — Built for Colab, SSH, or headless environments, printing tensor stats step-by-step, displaying live ASCII charts for scalars, and supporting optional labeled PDF snapshots. Supports pausing training so you can tag new matrices or ask the AI to interpret results, right from the terminal.
Universal Backend Support — Automatically detects and works with NumPy, PyTorch, TensorFlow, CuPy, and JAX via a shared backend abstraction layer.
High Performance — Keeps overhead low by converting tensors to host-side NumPy arrays, reusing Matplotlib figures (set_data) instead of rebuilding them every step, and matching render sizes to the actual on-screen thumbnail.
Pulse AI Analyst — A context-aware chat panel briefed on its role that can inspect live matrix statistics, heatmaps, and your training code when "Send Code" is enabled.

Past Debugs

Debugged a custom LLM after a vocab size increase (2.5x) by catching a normalization bug — dividing residual growth by math.sqrt(num_layers) instead of num_layers — that let activations blow up and halted training.
Debugged another developer's custom attention mechanism producing NaN loss, tracing it to a missing infinity check before a division.

See pulsedb.netlify.app for screenshots of matrix selection, the live dashboard, and the CLI view.

Install

pip install pulseml

tkinter is required for GUI mode and ships with most Python installs. On Debian/Ubuntu, if it's missing:

sudo apt install python3-tk

For CLI-mode PDF snapshots, fpdf2 is installed automatically as part of the base package.

Quickstart

Import auto_track and call it right before your training loop starts. Make sure your loop is wrapped in if __name__ == '__main__':.

from pulse import auto_track

if __name__ == '__main__':
    auto_track()   # pass your training function for shape discovery, or call directly

    # Your training loop
    for epoch in range(num_epochs):
        # Training logic here
        pass

AI Chat & API Keys

To enable the AI chat panel, set the relevant provider's API key as an environment variable (e.g. ANTHROPIC_API_KEY, OPENAI_API_KEY, GEMINI_API_KEY, DEEPSEEK_API_KEY) — or leave it unset and Pulse will prompt you for one inside the GUI the first time you send a message.

License

Proprietary. See LICENSE. Use of this software is governed by the terms in that file — copying, redistribution, and reverse engineering are not permitted.

Metadata

Release files for pulseml 0.2.18

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for pulseml 0.2.18
File	Size	Uploaded
pulseml-0.2.18.tar.gz	459.3 kB	Sep 22, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for pulseml 0.2.18
File	Interpreter	ABI	Platform	Reset
pulseml-0.2.18-py3-none-any.whl 422.4 kB Sep 22, 2026	Python 3	none	any	Details

Total release size: 881.7 kB

Release files / pulseml-0.2.18.tar.gz

Download URL	pulseml-0.2.18.tar.gz
Size	459.3 kB
Tags	Source
SHA-256 checksum How to use checksums	`7268602df7e08ebccca5be3b36cd26836bd17ae0ceebd12c2ac4c0f610b2667a`
BLAKE2b-256 checksum How to use checksums	`af9fdc1dce0e7bb59a04d0b92698132cadc057d46d1d5e96f163eb01b1027d3c`
Upload date	Sep 22, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.11.9`

Release files / pulseml-0.2.18-py3-none-any.whl

Download URL	pulseml-0.2.18-py3-none-any.whl
Size	422.4 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`24b492854842c0509ecf98e6f78c857f8b64bb2ca773fd8c20aa8e242ce5d1db`
BLAKE2b-256 checksum How to use checksums	`1d6a31510c171ac14627ddd943e2ecf7646fd926c8c4678e9193859d95b0df03`
Upload date	Sep 22, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.11.9`