Umar

icon company Saarland University icon location Saarbrücken, Germany icon location Computer Vision | Machine Learning | Python | C++

Results 3 issues of Umar

More than 1 `input_shape` when initializing `flax_module`

3 comment

Some modules require more than 1 input when initializing, which can be passed through `kwargs`. But this doesn't work in some cases. For example: ```python class RNN(nn.Module): @functools.partial( nn.transforms.scan, variable_broadcast='params',...

enhancement
good first issue

Add bash completion script which can be installed/uninstalled with Makefile

Resolves #19

Fix command to create conda environment from `environment.yml`