-
Jraph uses `jax.tree_multimap`, which is deprecated and causes our tests to fail. Therefore I added a filterwarning in `pytest.ini`. We should remove this filter warning once Jraph fixes this.
-
### Issue Type
Bug
### Modules Involved
Documentation/Tutorial/Example
### Have you reproduced the bug with SPU HEAD?
Yes
### Installation Kind
binary
### SPU Version
spu 0.4.1b1
### OS Plat…
-
This might be related to #197, but I don't have enough knowledge to confirm it.
In short, while using `optax` with `flax.linen`, `optimizer.update` forces jit-recompilation of the `integer_pow` jax…
-
Implement optax-like optimizer for CMGF update step
-
### Describe your issue.
I have tried use dendrogram from Kaggle notebook but get error:
```
---------------------------------------------------------------------------
ImportError …
-
## Issue description
Build of `python39Packages.dm-tree` failed on x86_64-linux as of 2da37482b1c5d99851dc683c714429734cba4191. This is currently breaking `python39Packages.flax`.
```
error: builder…
-
- [x] I have checked that this issue has not already been reported.
- [x] I have confirmed this bug exists on the latest version of scanpy.
- [ ] (optional) I have confirmed this bug exists on the m…
-
**Describe the bug**
Got this error: TypeError: DataModule.__init__() got an unexpected keyword argument 'dataset', even though the docs stated that it's part of the parameters.
**To Reproduce**
…
-
Hi,
I tried following the example in https://github.com/google/flax/discussions/1876#discussioncomment-2143866 and ran into an ModuleNotFoundError. I was able to fix it by install tensorflow, but w…
-
### Bug description
`all_gather` function somehow produces negative values.
Here is the code snippet I have in `on_train_epoch_end` function:
```
if torch.any(self.query_labels < 0):
…