Skip to content
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

[Discussion] Sequencing side-effects in JAX #69

Open
vlainic opened this issue Aug 3, 2023 · 0 comments
Open

[Discussion] Sequencing side-effects in JAX #69

vlainic opened this issue Aug 3, 2023 · 0 comments

Comments

@vlainic
Copy link

vlainic commented Aug 3, 2023

Sequencing side-effects in JAX is a known issue in JAX. I have tried to make a custom operational env/task where the step update needs to be sequential.

JAX docs say one needs to tokenize the functions to force the sequence. However, I could not find any task in EvoJax with the tokens. Did anybody ever try that? Or I should think more and redesign my problem to make it more JAX compatible since I use a bunch of jax.lax.cond (constraints are quite complicated for the problem I have).
@vlainic vlainic changed the title Sequencing side-effects in JAX [Discussion] Sequencing side-effects in JAX Aug 4, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@vlainic