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

Test all funtions with jax.test_util.check_grads to ensure differentiability #438

Open
paddyroddy opened this issue Nov 19, 2024 · 0 comments
Open

Test all funtions with jax.test_util.check_grads to ensure differentiability #438

paddyroddy opened this issue Nov 19, 2024 · 0 comments
Assignees
@paddyroddy @Saransh-cpp
Labels
array-api Work is related to the Array API enhancement New feature or request testing Work is related to testing

Comments

@paddyroddy
Copy link
Member

Is Your Feature Request Related to a Problem? Please Describe

As we move towards porting stuff to the Array API to make things GPU enabled, we ultimately one them to be differentiable too. Rather than coming up with a solution that works for GPU, which then needs a massive rewrite for differentiability. It would be good to ensure functions are on route to differentiability at the time.

Describe the Solution You'd Like

No response

Describe Alternatives You've Considered

No response

Additional Context

No response
@paddyroddy paddyroddy added array-api Work is related to the Array API enhancement New feature or request testing Work is related to testing labels Nov 19, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@paddyroddy paddyroddy

@Saransh-cpp Saransh-cpp

Labels
array-api Work is related to the Array API enhancement New feature or request testing Work is related to testing
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@paddyroddy @Saransh-cpp