Ecosyste.ms: Issues

An open API service for providing issue and pull request metadata for open source projects.

GitHub / google-deepmind/dm-haiku issues and pull requests

#808 - Remove usage of jax2tf.PolyShape in haiku.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#805 - Update references to JAX's GitHub repo

Pull Request - State: closed - Opened by copybara-service[bot] 2 months ago

#804 - WIP: Avoid pmap issue with linear layer in haiku.

Pull Request - State: closed - Opened by copybara-service[bot] 3 months ago

#798 - Cut 0.0.14.dev release.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#797 - Cut 0.0.13 release.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#791 - [Haiku] Add shard_map flops estimate to the jaxpr_info.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago - 1 comment

#788 - Integrating vmap with BatchNorm

Issue - State: open - Opened by reemabdelrazek30 7 months ago

#787 - Relax error message in statful_tests

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago

#786 - hk.BatchNorm with jax.vmap

Issue - State: open - Opened by reemabdelrazek30 7 months ago

#782 - Make flax an optional dependency.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#781 - Bump required JAX and TF versions to latest.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#780 - Consider make flax an optional dependency

Issue - State: closed - Opened by jjyyxx 9 months ago - 1 comment

#778 - Wrapping the ```init``` function inside ```jax.jit```

Issue - State: closed - Opened by ksmdnl 9 months ago - 1 comment

#777 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#775 - Exposing the experimental _split_transpose JAX scan parameter in Haiku.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago - 1 comment

#774 - TypeError: 'type' object is not subscriptable

Issue - State: closed - Opened by chenyzh28 11 months ago - 4 comments

#772 - Fix some JAX key reuse issues in haiku tests

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#771 - Cut 0.0.12 release.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#770 - Cut 0.0.13.dev release.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#768 - Use tf-nightly for Haiku testing.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#767 - Deactivate failsafe in hk.flax.lift to avoid false-positives.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#766 - Train multiple hk.nets.MLP with one optimizer

Issue - State: closed - Opened by rsmath about 1 year ago - 2 comments

#765 - modify layer_stack transparency map

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#764 - Fix _compute_fans when fan_in_axes is specified and len(shape) is 2.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#763 - Haiku: fix jaxpr_info_test for jax v0.4.24

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#762 - Skip broken test.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#761 - enable state with layer stack

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#760 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#758 - Add a comment to force_name to avoid overuse.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#757 - Cut 0.0.12.dev release.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#756 - Cut 0.0.11 release.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#755 - `haiku.experimental.flax` is not part of newest pip release

Issue - State: closed - Opened by SimiPixel over 1 year ago - 1 comment

#753 - Fix packaging issue

Pull Request - State: open - Opened by ethanluoyc over 1 year ago

#752 - Issue with wheels including docs and examples folder

Issue - State: open - Opened by ethanluoyc over 1 year ago

#750 - Direct Feedback Alignment

Issue - State: open - Opened by ferqui over 1 year ago

#749 - Do not run doctests as part of the docs configuration

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#748 - Updated the `hk.mixed_precision.set_policy` docstring

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#747 - haiku: avoid calls to deprecated jax.random APIs

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#746 - Install requirements jointly

Pull Request - State: closed - Opened by superbobry over 1 year ago

#744 - haiku: avoid deprecated jax.random.default_prng_impl()

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#743 - Updated jax.config import

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#742 - Updated jax.config import

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#741 - Wrong gradients in a Haiku network

Issue - State: open - Opened by SNMS95 over 1 year ago

#740 - Modules got silently "reused" with `hk.vmap`

Issue - State: open - Opened by jjyyxx over 1 year ago - 2 comments

#737 - 0.4.16 timeline

Issue - State: closed - Opened by joaogui1 over 1 year ago - 2 comments

#736 - haiku: fully support JAX typed PRNG keys

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#734 - Add error message to the layer stack assertion.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#733 - Fixed a typo in the parameter sharing notebook

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#732 - Ran pyupgrade --py39-plus **/*.py

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#731 - Run integration tests on 3.9-3.11

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#729 - Do not test on 3.8, since we no longer support it

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#728 - Avoid crashes when constructing signatures

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#727 - Do not use ... as a type

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#726 - ModuleMetaclass now directly subclasses abc.ABCMeta

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#725 - Dropped 3.8 from the list of supported versions

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#723 - How to use `apply` with additional parameters?

Issue - State: closed - Opened by Lando-L over 1 year ago - 1 comment

#716 - Fix Pyright error when creating modules

Pull Request - State: closed - Opened by NeilGirdhar over 1 year ago

#703 - Add .editorconfig

Pull Request - State: closed - Opened by NeilGirdhar over 1 year ago

#699 - Don't install requirements-test in docs CI.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#690 - Edit hints and comments mentioning `jax.scan()` to `jax.lax.scan()`

Pull Request - State: closed - Opened by paulbricman over 1 year ago - 1 comment

#678 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#677 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#663 - Document Typo

Issue - State: closed - Opened by keep9oing over 1 year ago

#659 - Allow generating split params from layer_stack.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#645 - rnn classifies mnist

Issue - State: open - Opened by never-to-never almost 2 years ago - 3 comments

#579 - BatchNorm No value for 'average' ValueError, perhaps set an init function?

Issue - State: closed - Opened by noahadhikari about 2 years ago - 2 comments

#560 - Delete `jax_experimental_name_stack` flag

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#552 - Re-running the training produces very different result

Issue - State: closed - Opened by osbm over 2 years ago - 1 comment

#521 - Fix typo "example"

Pull Request - State: closed - Opened by david-lindner over 2 years ago - 2 comments

#517 - Preserve Haiku overriden docstring for `hk.cond`.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago