issues
search
google
/
objax
Apache License 2.0
769
stars
77
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Give error if assigning variable wrong shape
#115
carlini
closed
3 years ago
4
"objax.variable.VarCollection is not a valid JAX type" when creating a custom optimizer
#114
RXZ2020
closed
3 years ago
11
Developer documentation and guide on adding new code
#113
AlexeyKurakin
closed
4 years ago
1
Document how to set random seed for default generator
#112
peterjliu
closed
3 years ago
1
Remove need for .value when referring to internal param values.
#111
peterjliu
opened
4 years ago
1
Initial dot product attention
#110
joaogui1
closed
3 years ago
12
Explore Objax to Tensorflow conversion
#109
AlexeyKurakin
closed
3 years ago
5
Add CIFAR10 Colab tutorial
#108
AlexeyKurakin
closed
4 years ago
3
Add requirements.txt for tests
#107
AlexeyKurakin
closed
4 years ago
0
Add type and shape check to var assign
#106
rwightman
closed
3 years ago
3
Updated instructions for installing jaxlib.
#105
david-berthelot
closed
4 years ago
0
Many layer conventions are different from both Tensorflow/Keras and PyTorch
#104
rwightman
opened
4 years ago
4
Fix parallel test cases to allow floats to be almost zero
#103
carlini
closed
4 years ago
0
Modify ResNet_V2 implementation to be compatible with Keras implementation.
#102
kihyuks
closed
3 years ago
4
Give better error message when calling Parallel() without replicate()
#101
carlini
closed
3 years ago
7
Add a test for random number generator.
#100
shs037
closed
4 years ago
0
Give better error when passing wrong input shape to convolution
#99
carlini
closed
4 years ago
0
Pooling padding.Resolves #91
#98
david-berthelot
closed
4 years ago
1
Move RNN to layers.py and make it stateless.
#97
aterzis-google
opened
4 years ago
3
Add super convergence example
#96
sathish-a
opened
4 years ago
1
Add warning message if restoring from empty path
#95
carlini
closed
4 years ago
0
Initialize layer-norm properly.
#94
peterjliu
closed
4 years ago
0
Override default beta params in Adam
#93
sathish-a
closed
4 years ago
0
Override default momentum
#92
sathish-a
closed
4 years ago
0
Padding for pooling layers
#91
kihyuks
closed
4 years ago
0
Computing gradients for a generator and discriminator
#90
bilal2vec
closed
4 years ago
4
Clarification on loss function names
#89
bilal2vec
closed
4 years ago
1
Tracer error when using a random variable
#88
bilal2vec
closed
4 years ago
5
Release branch for 1.1.0
#87
david-berthelot
closed
4 years ago
0
Minor changes to documentation.
#86
david-berthelot
closed
4 years ago
0
Naming manipulation tools to facilitate importing model weights from …
#85
david-berthelot
closed
4 years ago
1
Documentation for ForceArgs
#84
AlexeyKurakin
closed
4 years ago
0
[Proposal] Orthogonal Initializer
#83
joaogui1
closed
4 years ago
2
[Proposal] Identity Initializer
#82
joaogui1
closed
4 years ago
2
Formatting fix for objax.privacy.dpsgd package.
#81
aterzis-google
closed
4 years ago
0
Allow strings and numerical values for Conv2D/ConvTranspose2D padding.
#80
david-berthelot
closed
4 years ago
0
Clarify JIT docs about variables passed for a module.
#79
david-berthelot
closed
4 years ago
0
Show JaxArray type on readthedocs.
#78
david-berthelot
closed
4 years ago
0
Attempt to fix https://github.com/google/objax/issues/72
#77
aterzis-google
closed
4 years ago
0
Fix documentation error in flatten.
#76
david-berthelot
closed
4 years ago
0
Remove type hints.
#75
aterzis-google
closed
4 years ago
1
Add signature support to wrappers, add Function module.
#74
david-berthelot
closed
4 years ago
0
Adding ForceArgs module
#73
AlexeyKurakin
closed
4 years ago
2
Sphinx documentation looks different locally and on readthedocs.io
#72
david-berthelot
closed
4 years ago
0
Cleanup and reformat examples directory
#71
AlexeyKurakin
closed
3 years ago
3
Enable ConvTranpose2D groups argument
#70
david-berthelot
opened
4 years ago
2
Add LSTM cell to layers
#69
david-berthelot
opened
4 years ago
5
Write a GAN example (maybe port BigGAN or something similarly popular)
#68
david-berthelot
opened
4 years ago
8
Create example on gradient accumulation
#67
david-berthelot
opened
4 years ago
0
Extend GPT-2 example to have a chat discussion and make it fast
#66
david-berthelot
opened
4 years ago
0
Previous
Next