Changelog

All notable changes to this project will be documented in this file.

The format is based on Keep a Changelog, and this project adheres to Semantic Versioning.

Unreleased

v0.2.0 - 2025-07-10

Added

• .train_on_batch() and .fit_on_batch() methods to ImpactModel (@markean, #15).
• Custom ArrayDataset class for handling data in ImpactModel, removing the need for the jax-dataloader dependency (@markean, #14).
• GitHub Pages documentation site (@markean, #10).
• Installation instructions in the documentation site (@markean, #10).
• ArrayLoader class supports shuffle and drop_last parameters for epoch training for .fit() (@markean, #15).

Changed

• Adopted jax.typing module for improved type hints.
• Removed unnecessary JAX array type conversion in ImpactModel methods.
• The .fit() method now uses epoch-based (minibatch) training (@markean, #15).
• Updated .fit(), .train_on_batch(), and .fit_on_batch() to train the model using the internal SVI state, continuing from the last state if available (@markean, #15).

Removed

• Removed the jax-dataloader dependency (@markean, #14).
• Removed the guide property, as it is part of the vi property.

v0.1.0 - 2025-06-27

Added

• Initial public release.