Releases: UW-CTRL/stljax
Releases · UW-CTRL/stljax
v1.1.0-tests-examples
masking operation for temporal operators
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 thattime_dim=0
- With masking, the need to reverse signals is removed.
jax implementation of original stlcg
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