google-deepmind / google-deepmind/chex
Error with Pydantic
Open
- Dominant language
- Python
- Stars
- 957
- Forks
- 74
- Avg merge
- 21h 10m
- Merged PRs (30d)
- 1
Description
Hello!
I'm interested in using pydantic's recursive constructor / asdict functionality, but jax.jit-ed functions give the following error:
```
Argument '_Pydantic_OptimConfig_93971134241088(.. SOMETHING HERE...)' of type is not a valid JAX type.
```
Contributor guide
Assessment
This issue has not been assessed yet.