Skip to content

jax implementation of original stlcg

Pre-release
Pre-release
Compare
Choose a tag to compare
@karenl7 karenl7 released this 29 Sep 01:38
· 47 commits to main since this release
v0.0.0
2c7f3a5
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

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

Assets 2
Loading