GitLab

* Fix env variable name in test.sh scripts to properly test pure-JAX implementations
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* Update test scripts to use pure-JAX impl in encoder

test_custom_call_compute.py already uses pure-JAX impl as
reference so testing the pure-JAX impl against itself would be
redundant. The encoder tests have their own implementation so
testing the pure-JAX impl of primitives is still useful.
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* Update qa/L0_jax_unittest/test.sh
Co-authored-by: Phuong Nguyen <phuonguyen@nvidia.com>
Signed-off-by: jberchtold-nvidia <158520091+jberchtold-nvidia@users.noreply.github.com>

---------
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
Signed-off-by: jberchtold-nvidia <158520091+jberchtold-nvidia@users.noreply.github.com>
Co-authored-by: Phuong Nguyen <phuonguyen@nvidia.com>