Avoid passing concrete argument to jax.remat
#3472
copybara-service[bot]flax_test.yml
on: push
Matrix: Test import standalone
Matrix: Run Tests
Annotations
1 error
|
Test pre-commit hooks
Process completed with exit code 1.
|