-
Currently hface environment is set up as follows in `CausalImage_ImageModelBackbones.R`
```
conda create -n hface python=3.11
conda activate hface
python3 -m pip install --upgrade tran…
-
[optax](https://optax.readthedocs.io/en/latest/) is the most popular JAX library for optimizers. Feature request: Let users pass an [optax.GradientTransformation](https://optax.readthedocs.io/en/lates…
-
- [ ] For most schedules, the total number of steps is specified through the `transition_steps` parameter, but in some cases (e.g., `optax.cosine_decay_schedule`, `optax.warmup_cosine_decay_schedule` …
-
```
ERROR: Cannot install flax because these package versions have conflicting dependencies.
The conflict is caused by:
optax 0.1.2 depends on jaxlib>=0.1.37
optax 0.1.1 depends on jaxli…
-
Hello,
I create a new Python environment and activate it.
Then I try pip install dalle-mini
After a while it errors out with
```
INFO: pip is looking at multiple versions of optax to determin…
-
Hi, thank you very much for providing such brilliant work and code, which greatly facilitates my research. I would like to inquire about the specific versions of the packages listed in the requirement…
-
This is using T4 GPU runtime. All the installs ran successfully, but the second cell failed:
```
AttributeError Traceback (most recent call last)
[](https://localhost:8…
-
The conflict is caused by:
praxis 1.4.0 depends on tfds-nightly==4.8.3.dev202303280045
The user requested jax==0.4.26
jax[cuda12] 0.4.26 depends on jax 0.4.26 (from https://pypi.tuna.ts…
-
Hi @jenkspt, I have been learning nanoGPT and reproducing it in JAX from scratch. Your repo has been a very helpful reference.
I encountered an issue with `optax.apply_every` and thought you might …
-
I always get an out of memory error using optax.MultiSteps, even when every_k_schedule=1.
Using optax.apply_every(k=1) in a chain works fine.
```
optimizer = optax.chain(
optax.clip_by_globa…