Issues

• how to save the values of gradient to the file?

  google/objax#272 · Filter by author Sherrylife opened on Dec 20, 2024

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

  google/objax#270 · Filter by author GaetanLepage opened on May 7, 2024

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

  google/objax#260 · Filter by author cyugao opened on Jun 12, 2023

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

  google/objax#254 · Filter by author alvarobartt opened on Dec 26, 2022
  1

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

  google/objax#253 · Filter by author alvarobartt opened on Dec 26, 2022
  1

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

  google/objax#245 · Filter by author a1302z opened on Aug 22, 2022
  3

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

  google/objax#234 · Filter by author gkaissis opened on Feb 7, 2022
  4

  

• RecursionError when attempting to unpickle objax objects

  google/objax#221 · Filter by author bytbox opened on May 7, 2021
  2

• Regression of JAX duck typing.

  bug
  google/objax#220 · Filter by author AlexeyKurakin opened on May 4, 2021
  2

  

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

  google/objax#219 · Filter by author shenzebang opened on May 1, 2021
  3

• Model compiling twice when using jax==0.2.10 or later

  google/objax#212 · Filter by author wil-j-wil opened on Mar 29, 2021
  6

• Improve automatic variable tracing

  google/objax#206 · Filter by author AlexeyKurakin opened on Mar 12, 2021

• Jit error for static_argument in a custom jax function

  google/objax#204 · Filter by author srxzr opened on Feb 24, 2021
  1

• objax.Vectorize batch_axis argument for multiple arguments default must be adaptively sized.

  google/objax#191 · Filter by author JeremyNixon opened on Jan 12, 2021

• Create learning rate schedulers

  feature requestquestion
  google/objax#187 · Filter by author david-berthelot opened on Dec 18, 2020
  1

  

• Objax2Tf enhancements

  google/objax#179 · Filter by author AlexeyKurakin opened on Dec 15, 2020

• Fix broken example: examples/text_generation/shakespeare_rnn.py

  google/objax#172 · Filter by author david-berthelot opened on Dec 14, 2020
  

• Cannot run imagenet_pretrained_vgg example

  google/objax#160 · Filter by author RXZ2020 opened on Nov 21, 2020
  5

  

• Calling objax.Grad twice on a model with BatchNorm with respect to input params crashes

  bugurgent
  google/objax#158 · Filter by author carlini opened on Nov 21, 2020
  2

  

• Improve error when recursively calling Parallel

  google/objax#157 · Filter by author carlini opened on Nov 21, 2020
  4

• Improve error when calling replicated function without Parallel

  google/objax#155 · Filter by author carlini opened on Nov 21, 2020
  1

• Need better error message for max_pool_2d

  google/objax#154 · Filter by author david-berthelot opened on Nov 20, 2020

• Calling objax.random.* from within a jitted function breaks the DEFAULT_GENERATOR

  feature requesturgent
  google/objax#147 · Filter by author carlini opened on Nov 18, 2020
  1

  

• Undocumented API load_pretrained_weights_from_keras

  google/objax#131 · Filter by author david-berthelot opened on Oct 28, 2020
  

• Guide for porting models from PyTorch

  google/objax#130 · Filter by author aterzis-google opened on Oct 27, 2020