Skip to content

Releases: UW-CTRL/stljax

v1.1.0-tests-examples

27 Nov 01:23
ff13b31
Compare
Choose a tag to compare

Major tidying of code base, bug fixes, adding test scripts and example notebooks.

masking operation for temporal operators

29 Sep 07:11
4be224d
Compare
Choose a tag to compare

Big update from the original stlcg version using recurrent graphs to manage temporal operations.

  • removed the need for input to have a batch dimension. batching is now handled by user via vmap. It is assumed that time_dim=0
  • With masking, the need to reverse signals is removed.

jax implementation of original stlcg

29 Sep 01:38
2c7f3a5
Compare
Choose a tag to compare
Pre-release

A direct JAX implementation of the original stlcg framework originally implemented in PyTorch.

Paper: https://arxiv.org/abs/2008.00097

Original PyTorch implementation: https://github.com/StanfordASL/stlcg/tree/dev