-
Phuong Nguyen authored
* softmax custom calls with correct encapsulates * rm jax deprecated features --------- Signed-off-by:Phuong Nguyen <phuonguyen@nvidia.com>
1975ace4
* softmax custom calls with correct encapsulates
* rm jax deprecated features
---------
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>