google-deepmind/chex

Issues

• Is my `reverse_cumsum` a good fit here?

  #365 opened a month ago by cool-RR
  2

• Use of deprecated type hints - PEP 585

  #361 opened 3 months ago by jeertmans
  0

• `chex.dataclass` wrapper causes type error: Expected no arguments to dataclass constructor

  #321 opened a year ago by carlosgmartin
  1

• ImportError: cannot import name 'DeviceLocalLayout' from 'jax._src.layout' (/opt/conda/lib/python3.10/site-packages/jax/_src/layout.py)

  #355 opened 7 months ago by maming109
  2

• AttributeError: module 'jax' has no attribute '_src'

  #197 opened 2 years ago by jenkspt
  4

• Test files are in binary package distribution

  #347 opened 8 months ago by daskol
  0

• Breaking for jax 0.4.24

  #332 opened 10 months ago by Intron7
  0

• Citation link broken

  #329 opened a year ago by felixchalumeau
  0

• Tuple not recognized as a valid chex.ArrayTree

  #323 opened a year ago by mblondel
  3

• Allow for nested chex.chexify

  #306 opened a year ago by nicow-elia
  1

• `jax.random.key` tree comparison results in a `ZeroDivision` warning

  #318 opened a year ago by joeryjoery
  2

• Next release

  #313 opened a year ago by GaetanLepage
  2

• TypeError: non-default argument 'value' follows default argument

  #305 opened a year ago by George-Ogden
  1

• Numpy Conflict

  #298 opened a year ago by nickknyc
  0

• Mypy index type error with `chex.dataclass`

  #296 opened a year ago by hylkedonker
  0

• conda package dependencies

  #270 opened a year ago by willis-richard
  5

• Strange gap in version

  #287 opened a year ago by daskol
  1

• `assert_tree_shape_prefix` requires tuple instead of `Sequence`

  #274 opened 2 years ago by hylkedonker
  2

• Support wrapping functools.partial() objects

  #262 opened 2 years ago by danijar
  1

• Missing package dependency typing-extension

  #259 opened 2 years ago by hylkedonker
  1

• `isinstance(None, (int, float, chex.Array))` raises error since `chex==0.1.7`

  #261 opened 2 years ago by vwxyzjn
  1

• Analog to `flax.struct.dataclass`

  #248 opened 2 years ago by homerjed
  2

• chex.Dimensions API enhancement

  #231 opened 2 years ago by wbrenton
  1

• ModuleNotFoundError: No module named 'jax.numpy'

  #226 opened 2 years ago by vballoli
  5

• post_init error in inherited dataclass

  #212 opened 2 years ago by thomaspinder
  1

• chex.disable_asserts() is ignored by chex.assert_max_traces

  #210 opened 2 years ago by Pascal314
  0

• Specify non-pytree node dataclass fields

  #64 opened 4 years ago by ethanluoyc
  3

• `AssertsChexifyTest.test_uninspected_checks` test failure

  #204 opened 2 years ago by samuela
  3

• Dill pickling chex.dataclass blows the stack

  #188 opened 2 years ago by JonyEpsilon
  0

• Typing issue with chex.dataclass

  #155 opened 3 years ago by wookayin
  2

• How to version?

  #166 opened 2 years ago by temirovazat
  1

• Better error report for max traces exceeded

  #176 opened 2 years ago by uduse
  2

• Using variants with pytest

  #163 opened 3 years ago by pablo2909
  2

• No attribute 'KeyArray' when importing chex

  #158 opened 3 years ago by raymondchua
  1

• CpuDevice no longer in jax

  #156 opened 3 years ago by adamgayoso
  4

• fake_pmap_and_jit has a confusing interface

  #144 opened 3 years ago by marcvanzee
  0

• chex.variants(with_pmap=True) ignores `static_argnames`

  #115 opened 3 years ago by fvisin
  9

• Consider supporting static attributes in chex.dataclass

  #104 opened 3 years ago by NeilGirdhar
  1

• Error with Pydantic

  #71 opened 3 years ago by kaiwenw
  0

• Dataclass breaks is_leaf function for jax.tree_map

  #62 opened 4 years ago by louiskirsch
  1

• [REQ] Conda recipe

  #37 opened 4 years ago by PhilipVinc
  2

• without_jit=True for already jitted functions

  #22 opened 4 years ago by fabiannagel
  0

• Chex dataclass throws an exception in Python 3.9

  #10 opened 4 years ago by dionhaefner
  1

• 'jax.interpreters.xla' has no attribute '_DeviceArray' for jax <= 0.2.5

  #9 opened 4 years ago by balancap
  0

• Chex dataclass defaulting mappable_dataclass=True

  #8 opened 4 years ago by balancap
  2

• Fake contexts by calling .start() not working

  #3 opened 4 years ago by sebamenabar
  2

• DeprecationWarning for importing toolz

  #4 opened 4 years ago by NeilGirdhar
  1