George Necula
5fabd34e7e
[jax2tf] Remove non-native serialization test from jax_to_ir_test
...
PiperOrigin-RevId: 683124315
2024-10-07 04:21:38 -07:00
Vladimir Belitskiy
6127baa117
Ignore the Deprecation warning produced about native_serialization=False.
...
PiperOrigin-RevId: 657221363
2024-07-29 09:11:54 -07:00
Sergei Lebedev
56745818a6
Added basic support for int2/uint2 dtypes to JAX
...
#21369
PiperOrigin-RevId: 649366888
2024-07-04 04:13:24 -07:00
Sergei Lebedev
a13efc2815
Added int4 and uint4 to dtype-specific tests
...
I probably missed some cases, so this PR is really just the first step in
making sure we have good *int4 coverage.
2024-04-18 15:20:20 +01:00
Sharad Vikram
74b136e62c
Delete jax_experimental_name_stack flag
...
PiperOrigin-RevId: 487601864
2022-11-10 11:59:50 -08:00
Peter Hawkins
ba557d5e1b
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
...
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
2022-09-22 12:27:19 -07:00
jax authors
840c96692e
Internal change
...
PiperOrigin-RevId: 468509799
2022-08-18 11:39:07 -07:00
Peter Hawkins
1e241dcf16
Catch ModuleNotFoundError instead of ImportError.
...
We frequently use the pattern
try:
import m
except ImportError:
# do something else.
This suppresses errors when the module can be found but does not import
successfully for any reason. Instead, catch only ModuleNotFoundError so
missing modules are allowed but buggy modules still report errors.
2022-08-18 15:22:49 +00:00
Sharad Vikram
5ff2e8eb4c
Fix name stack bugs
2022-04-19 11:14:41 -07:00
Tom Hennigan
2a6147af1b
Add build time support for AOT compilation to TF graphs.
...
PiperOrigin-RevId: 417392920
2021-12-20 06:18:39 -08:00
