Skip to content

Jax Bug Fix #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
jeappen wants to merge 23 commits into
base: master
Choose a base branch
from
Open

Jax Bug Fix #2

jeappen wants to merge 23 commits into from

Conversation

@jeappen
Copy link
Contributor

@jeappen jeappen commented Mar 21, 2024

  • Fix Predicate return value in jax (torch.min and jnp.min have different semantics)
  • Edit tests to enable running all tests at once (need to use importlib.reload to refresh loaded modules based on environment variable)

jeappen and others added 23 commits March 20, 2024 11:45
@jeappen
fix(jax): rename os env, reload module to make tests work together
18cdc0e
@jeappen
fix(jax): add evaluation shape test (which is failing). TODO: Debug
47e75ae
@jeappen
fix(jax): fix evaluation bug in predicates, jnp.min and torch.min had…
885e4d3 
… different semantics. Fix examples to allow different batch sizes as input to STL.eval
@jeappen
fix(jax) : change a few more numpy functions -> jax.numpy
677ba06
@jeappen
docs(jax) : update jax instructions and changelog
b9a615d
move common fns to utils so torch is optional
e007146
@jeappen
fix(stl): fix for using StlPySolver with JAX (temporary conversion to…
35a2cd0 
… numpy)
@jeappen
docs(stl): add comments to STLPy tools
8e5bf1e
@jeappen
feature(stl): add loop and branch specs. Fix always end_time.
74828db
@jeappen
feature(realrobot): add script to save output to npy
3a803ca
@jeappen
feature(stl): add loop tests and stlpy solver tests
b1848c8
@jeappen
Merge remote-tracking branch 'origin/feature/jax' into feature/jax
a2c47ac
@jeappen
fix(stlpy): add tests and fix stlpy conversion
1028bd4
@jeappen
feature(async): add vanish on end, show spec in pgf plot
51ee629
@jeappen
feature(stl): Working on JAX Until
f181b47
@jeappen
feature(stl): add single plan no formation
dd46986
@jeappen
Update README.md
6111114
@jeappen
fix(ja): towards STL PyTrees compatible with JAX (helpful for vmappin…
d89e4c6 
…g with JAX)
@jeappen
feature(jax): Allows multiple goal (in theory)
d3e1de6
@jeappen
fix(jax) : temp fix for RectPredicate not inheriting PredicateBase
f341758
@jeappen
feature(jax): add rich repr for center of pred
960a353
@jeappen
feature(jax): add tests towards fixing avoid not working for differen…
6922e51 
…tiable approaches (GCBF+ etc)
@jeappen
fix(jax): add tolerance value to tests
89b963c
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@jeappen