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.

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

Add jax.lax.reduce_window #12852

Closed
wants to merge 516 commits into from
Closed

Add jax.lax.reduce_window #12852

wants to merge 516 commits into from

Conversation

AnnaTz
Copy link
Contributor

@AnnaTz AnnaTz commented Mar 21, 2023

No description provided.

@AnnaTz
Copy link
Contributor Author

AnnaTz commented Mar 21, 2023

will have to add as_strided to ivy experimental in order to continue

@ivy-leaves ivy-leaves added the JAX Frontend Developing the JAX Frontend, checklist triggered by commenting add_frontend_checklist label Mar 21, 2023
@ivy-seed ivy-seed added the Stale label Mar 29, 2023
@ivy-seed ivy-seed closed this Apr 5, 2023
@AnnaTz AnnaTz reopened this Apr 5, 2023
@AnnaTz
Copy link
Contributor Author

AnnaTz commented Apr 5, 2023

will have to add as_strided to ivy experimental in order to continue

#13744

@ivy-llc ivy-llc deleted a comment from ivy-seed Apr 5, 2023
@ivy-llc ivy-llc deleted a comment from ivy-seed Apr 5, 2023
@AnnaTz AnnaTz removed the Stale label Apr 5, 2023
@AnnaTz AnnaTz linked an issue Apr 5, 2023 that may be closed by this pull request
@ivy-seed ivy-seed added the Stale label Apr 13, 2023
@ivy-seed ivy-seed closed this Apr 20, 2023
@AnnaTz AnnaTz reopened this Apr 20, 2023
@ivy-llc ivy-llc deleted a comment from ivy-seed Apr 20, 2023
@ivy-llc ivy-llc deleted a comment from ivy-seed Apr 20, 2023
@ivy-seed
Copy link

This PR has been labelled as stale because it has been inactive for more than 7 days. If you would like to continue working on this PR, then please add another comment or this PR will be closed in 7 days.

ivy-branch and others added 20 commits April 22, 2023 16:34
* refactored control flow ops and added docstrings
- Add `LocalIvyImporter` to `dynamic_import` for local import to work properly.
- Add `local_ivy_id` for local ivy id to be used inside `impersonate_import` to fetch the local object through global ivy. 
- Refactor `impersonate_import` implementation.
Add functionality for ivy.with_backend to skip compiling specific modules, save resources, and allow for type sharing e.g. (ivy.Array binds to the same class).
…jax.numpy.tril_indices and ensure passing when the front is set as the backend is set
* add functions for flax transpilation

* lint
…s and updated frontend __getitem__ methods to use it
@AnnaTz AnnaTz changed the base branch from master to array_container_ivy_binding April 22, 2023 15:35
@AnnaTz AnnaTz changed the base branch from array_container_ivy_binding to master April 22, 2023 15:36
@AnnaTz AnnaTz closed this Apr 22, 2023
@AnnaTz AnnaTz deleted the lax_reduce_window branch April 22, 2023 15:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
JAX Frontend Developing the JAX Frontend, checklist triggered by commenting add_frontend_checklist Stale
Projects
None yet
Development

Successfully merging this pull request may close these issues.

jax.lax.reduce_window