Issues
- 0
- 6
TrainVar with single scalar causing error
#164 opened by matpalm - 2
Missing release 1.7.0 on GitHub
#263 opened by GaetanLepage - 0
TypeError during gradient computation: type <class 'objax.variable.TrainVar'> is not a valid JAX type
#260 opened by cyugao - 1
- 1
`objax.variable.VarCollection.update` not compliant with key-value assignment
#254 opened by alvarobartt - 3
pmean inside objax.parallel causes multithreading deadlock for more than 2 gpus
#245 opened by a1302z - 2
Regression of JAX duck typing.
#220 opened by AlexeyKurakin - 2
- 4
- 2
Closure scoping for nested objax.Functions
#237 opened by mathDR - 3
replacing jax.vmap with objax.Vectorize
#235 opened by mathDR - 0
Error due to the deprecation of jax.api
#228 opened by Min-Li - 3
How to compute Jacobian of outputs w.r.t. inputs
#224 opened by smao-astro - 1
- 0
Raise warning when arguments to loss functions are not of the same dimensions
#190 opened by aterzis-google - 2
- 3
- 2
Accessing variable of a vectorized module
#217 opened by noashin - 0
- 6
- 0
If user won't add random generator to VarCollection of jitted code then same number always return by random generator
#208 opened by AlexeyKurakin - 0
Improve automatic variable tracing
#206 opened by AlexeyKurakin - 0
- 1
Jit error for static_argument in a custom jax function
#204 opened by srxzr - 1
Create learning rate schedulers
#187 opened by david-berthelot - 4
Request: remove strict dependence on tensorflow
#199 opened by bytbox - 0
objax.Sequential stopped working with jax.numpy.pad in latest jax update
#192 opened by david-berthelot - 0
objax.Vectorize batch_axis argument for multiple arguments default must be adaptively sized.
#191 opened by JeremyNixon - 2
Calling objax.Grad twice on a model with BatchNorm with respect to input params crashes
#158 opened by carlini - 0
Objax2Tf in-depth guide or tutorial
#180 opened by AlexeyKurakin - 0
Improve BatchNorm numerical stability
#182 opened by david-berthelot - 10
- 0
Objax2Tf enhancements
#179 opened by AlexeyKurakin - 2
MNIST tutorial bigger model does not converge
#173 opened by david-berthelot - 0
Update documentation to prefer Function.with_vars decorator as it is less error prone.
#168 opened by david-berthelot - 0
- 0
- 0
- 2
Unclear how to use Dropout with Sequential
#162 opened by peterjliu - 0
Add warning: calling checkpoint.save() within a replicate() saves the replicated variables
#163 opened by carlini - 1
- 1
- 1
Calling objax.random.* from within a jitted function breaks the DEFAULT_GENERATOR
#147 opened by carlini - 5
Cannot run imagenet_pretrained_vgg example
#160 opened by RXZ2020 - 0
404 error for many Objax examples
#156 opened by RXZ2020 - 4
Improve error when recursively calling Parallel
#157 opened by carlini - 0
Need better error message for max_pool_2d
#154 opened by david-berthelot - 0
Add jax.lax.scan to functional.
#150 opened by david-berthelot - 2