tests/jax/test_custom_call_compute.py · c3b7c2aee161ccfebb9b37eea9b0afdd10e972c1 · OpenDAS / TransformerEngine

Phuong Nguyen authored Jun 12, 2025

Revert "[JAX] GroupedDense v.2 without dynamic shape (#1721)"

This reverts commit 5d01ef21

.
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

c3b7c2ae

test_custom_call_compute.py 56.6 KB