Comments (4)
The fix was included in v0.1.3
from chex.
We fixed this in chex in 7552dfc so yes we would need a new release if a new release of JAX has been made (cc @hbq1 ).
from chex.
Indeed 0.3.7 jax is out now https://pypi.org/project/jax/#history
Thanks!
from chex.
So, when is this going to be released? @tomhennigan thanks!
from chex.
Related Issues (20)
- Analog to `flax.struct.dataclass` HOT 2
- Missing package dependency typing-extension HOT 1
- `isinstance(None, (int, float, chex.Array))` raises error since `chex==0.1.7` HOT 1
- Support wrapping functools.partial() objects HOT 1
- conda package dependencies HOT 5
- `assert_tree_shape_prefix` requires tuple instead of `Sequence` HOT 2
- Strange gap in version HOT 1
- Mypy index type error with `chex.dataclass`
- Numpy Conflict
- TypeError: non-default argument 'value' follows default argument HOT 1
- Allow for nested chex.chexify HOT 1
- Next release HOT 2
- `jax.random.key` tree comparison results in a `ZeroDivision` warning HOT 2
- `chex.dataclass` wrapper causes type error: Expected no arguments to dataclass constructor HOT 1
- Tuple not recognized as a valid chex.ArrayTree HOT 3
- Citation link broken
- Breaking for jax 0.4.24
- Test files are in binary package distribution
- ImportError: cannot import name 'DeviceLocalLayout' from 'jax._src.layout' (/opt/conda/lib/python3.10/site-packages/jax/_src/layout.py) HOT 2
- Use of deprecated type hints - PEP 585
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from chex.