google-deepmind chex issues

google-deepmind / chex

https://chex.readthedocs.io

Apache License 2.0

769 stars 44 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

increase version number for a pypi release

#363 copybara-service[bot] closed 1 week ago
0
make `replacement` kwarg in warn_deprecated_function optional

#362 copybara-service[bot] closed 1 week ago
0
Use of deprecated type hints - PEP 585

#361 jeertmans opened 2 weeks ago
1
Sort devices explicitly by process index, then id (as opposed to IDs alone). IDs may be randomly generated, and are not guaranteed to be ordered based on their process index.

#360 copybara-service[bot] opened 2 months ago
1
test new CI config

#359 copybara-service[bot] closed 2 months ago
0
Chex: Add CI job for oldest supported JAX version

#358 copybara-service[bot] closed 3 months ago
0
Add CI job for oldest supported JAX version

#357 jakevdp closed 3 months ago
1
Disable registration of dataclasses defined in __main__ with JAX tree util.

#356 copybara-service[bot] closed 4 months ago
0
ImportError: cannot import name 'DeviceLocalLayout' from 'jax._src.layout' (/opt/conda/lib/python3.10/site-packages/jax/_src/layout.py)

#355 maming109 opened 4 months ago
2
Replace deprecated `jax.tree_*` functions with `jax.tree.*`

#354 copybara-service[bot] closed 4 months ago
0
Remove a bad async chexify test as the following is not guaranteed:

#353 copybara-service[bot] closed 4 months ago
0
Remove a bad async chexify test as the following is not guaranteed:

#352 copybara-service[bot] closed 4 months ago
0
Update and re-enable RTD.

#351 copybara-service[bot] closed 5 months ago
0
Test RTD webhooks.

#350 copybara-service[bot] closed 5 months ago
0
Fix linter's warnings.

#349 copybara-service[bot] closed 5 months ago
0
Fix typo.

#348 copybara-service[bot] closed 5 months ago
1
Test files are in binary package distribution

#347 daskol opened 5 months ago
0
Silence some pytype errors.

#346 copybara-service[bot] closed 6 months ago
0
Add utility to create a deprecated alias.

#345 copybara-service[bot] closed 6 months ago
0
Suppress not-callable pylint error for now since is being flagged erroneously all over the place.

#344 copybara-service[bot] closed 6 months ago
0
Fix doc-string in warn_deprecated_function.

#343 copybara-service[bot] closed 6 months ago
0
Increase version number to make a new release.

#342 copybara-service[bot] closed 6 months ago
0
Add flattening functionality to chex.Dimensions.

#341 copybara-service[bot] closed 6 months ago
0
Add wrapper for deprecating functions.

#340 copybara-service[bot] closed 6 months ago
0
Add Dimensions.size to get flat size from named shape.

#339 copybara-service[bot] closed 6 months ago
0
Add Dimensions.size to get flat size from named shape.

#338 copybara-service[bot] closed 6 months ago
0
Add warnings in chex.

#337 copybara-service[bot] closed 6 months ago
0
Register dataclass at definition so that unpickling the dataclass object, type, and treedef works in most cases (with one exception).

#336 copybara-service[bot] closed 7 months ago
0
#chex: Improve repr of dataclass keys/values.

#335 copybara-service[bot] closed 7 months ago
0
Adjust test tolerances for test_assert_trees_all_close

#334 copybara-service[bot] closed 7 months ago
0
Update `chex.assert_type` to check concrete types instead of just asserting that the type is a floating/integer sub-type.

#333 copybara-service[bot] closed 7 months ago
0
Breaking for jax 0.4.24

#332 Intron7 closed 7 months ago
0
Update test.sh

#331 copybara-service[bot] closed 7 months ago
0
Automated Code Change

#330 copybara-service[bot] closed 7 months ago
0
Citation link broken

#329 felixchalumeau opened 8 months ago
0
Restrict `assert_type` type checking when expected type is a dtype

#328 copybara-service[bot] closed 7 months ago
0
adding static_keynames support

#327 mishmish66 opened 9 months ago
2
Release 0.1.86.

#326 copybara-service[bot] closed 7 months ago
0
Avoid use of deprecated `device_buffer` attriutes of jax.Array

#325 copybara-service[bot] closed 10 months ago
0
Replace references to deprecated device_buffer attributes

#324 copybara-service[bot] opened 10 months ago
0
Tuple not recognized as a valid chex.ArrayTree

#323 mblondel closed 9 months ago
3
Use only 'jax_disable_jit' in `fake_jit`.

#322 copybara-service[bot] closed 10 months ago
0
`chex.dataclass` wrapper causes type error: Expected no arguments to dataclass constructor

#321 carlosgmartin opened 10 months ago
1
Release 0.1.85

#320 copybara-service[bot] closed 10 months ago
0
Install setuptools for python>=3.12.

#319 copybara-service[bot] closed 10 months ago
0
`jax.random.key` tree comparison results in a `ZeroDivision` warning

#318 joeryjoery closed 10 months ago
2
[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

#317 copybara-service[bot] closed 10 months ago
0
Improve usability of `chexify`.

#316 copybara-service[bot] closed 11 months ago
0
Silence `invalid-name` warning.

#315 copybara-service[bot] closed 11 months ago
0
Release 0.1.84.

#314 copybara-service[bot] closed 11 months ago
0