Skip to content

Conversation

ricardoV94
Copy link
Member

Closes #434

CC @kc611, am I missing a better solution?

@ricardoV94 ricardoV94 added bug Something isn't working numba labels Sep 11, 2023
@twiecki twiecki merged commit da66c2e into pymc-devs:main Sep 12, 2023
@ricardoV94 ricardoV94 changed the title Fix SoftmaxGrad failure with constant dy in numba backend Fix SoftmaxGrad failure with constant dy in numba backend Sep 12, 2023
@ricardoV94 ricardoV94 deleted the patch_numba_grad_softmax branch April 21, 2025 13:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working numba

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Numba SoftmaxGrad fails when input has constant shape

3 participants