-
### Description
I am interested in JAX-based optimization and would like to inquire about the potential implementation of second-order optimization methods such as Sequential Quadratic Programming (S…
-
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…
-
Hi,
I am trying to optimize a couple of parameters with different sizes and associated optimizers in a `jax.lax.while_loop` .
In order to achieve that, I used `multi_transform` and `set_to_zero…
-
Dropout is really the bane of equinox it seems. Loose follow-up of #681 - effectively, I'm trying to fix this problem that cropped up a while ago when using `optax.MultiSteps` for gradient accumulatio…
-
- [ ] 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` …
-
[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…
-
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…
-
```
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…
-
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…
-
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…