Matthew Johnson 9c39b6f70c update relu6 grad at 0 and 6 to match pytorch convention

2023-03-07 17:30:17 -08:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

functions.py

update relu6 grad at 0 and 6 to match pytorch convention

2023-03-07 17:30:17 -08:00

initializers.py

Replace jax._src.util.prod with math.prod.

2023-02-28 12:41:00 -08:00