← All projects
Training & fine-tuning

tensorboardX

Lets programmers log charts, images and sounds from AI training so they can watch progress in TensorBoard.

New this week

What it does

When you train an AI model, it helps to watch numbers like accuracy change over time. TensorBoard is a popular dashboard for this, and tensorboardX makes it easy to send data to it from PyTorch and other tools.

With simple function calls you can record numbers, images, charts, audio, text, model diagrams and more.

You then open TensorBoard to see everything as interactive graphs.

Who it's for

Machine learning developers and researchers who want to track their training runs.

Can I use it without coding?

Developers only. It is a Python library used inside training code. New to this? Start here.

Can I run it?

Cost
FreeNothing to pay or sign up for.
Computer
An ordinary computer is fine
Privacy
Runs on your own device: your data stays with you
API key
Not needed

Try it

Using Claude Code, Cursor or another AI coding assistant? Paste the prompt and it will walk you through the setup.New to this?

Before you install

  • Only download from the official GitHub page. Copies with similar names exist for many popular projects.
  • License (MIT): Very permissive: free to use, change and share, including for business.
  • More on staying safe