Search results

how to save the values of gradient to the file?

Status: Open.

#272 In google/objax;

· Sherrylife opened on Dec 20, 2024

[Jax 0.4.27] AttributeError: arr.device_buffers has been deprecated.

Status: Open.

#270 In google/objax;

· GaetanLepage opened on May 7, 2024

TypeError during gradient computation: type <class 'objax.variable.TrainVar'> is not a valid JAX type

Status: Open.

#260 In google/objax;

· cyugao opened on Jun 12, 2023

objax.variable.VarCollection.update not compliant with key-value assignment

Status: Open.

#254 In google/objax;

· alvarobartt opened on Dec 26, 2022

objax.variable.VarCollection.update fails when passing Dict[str, Any]

Status: Open.

#253 In google/objax;

· alvarobartt opened on Dec 26, 2022

pmean inside objax.parallel causes multithreading deadlock for more than 2 gpus

Status: Open.

#245 In google/objax;

· a1302z opened on Aug 22, 2022

objax.Jacobian and objax.Hessian similar to objax.Grad

Status: Open.

#234 In google/objax;

· gkaissis opened on Feb 7, 2022

RecursionError when attempting to unpickle objax objects

Status: Open.

#221 In google/objax;

· bytbox opened on May 7, 2021

Regression of JAX duck typing.

bugSomething isn't workingSomething isn't working

Status: Open.

#220 In google/objax;

· AlexeyKurakin opened on May 4, 2021

objax.Jit reports error when StateVar is added to the vc argument

Status: Open.

#219 In google/objax;

· shenzebang opened on May 1, 2021

Model compiling twice when using jax==0.2.10 or later

Status: Open.

#212 In google/objax;

· wil-j-wil opened on Mar 29, 2021

Improve automatic variable tracing

Status: Open.

#206 In google/objax;

· AlexeyKurakin opened on Mar 12, 2021