Skip to content

issues Search Results · repo:google-deepmind/jraph language:Python

Filter by

39 results
 (67 ms)

39 results

ingoogle-deepmind/jraph (press backspace or delete to remove)

google-deepmind/jraph
`GraphsTuple` vs. batched `GraphsTuple` in `get_get_graph_padding_mask`

Hey, thanks for this great package. I realized that jraph.get_graph_padding_mask returns jnp.array([False]) when applied to a non-batched GraphsTuple. I am wondering why this is? Would it be possible ...
  • thorben-frank
  • Opened 
    on Nov 8, 2024
  • #55

google-deepmind/jraph
HyperGraph support?

Hello. I was wondering if there is a plan to provide support for hyper graphs in jraph? The GraphsTuple class seems to be specifically designed for graphs and not easily adaptable to the case of hyper ...
  • errhernandez
  • Opened 
    on Aug 28, 2024
  • #53

google-deepmind/jraph
pad_with_graphs written with numpy

Hello! I was wondering if there is any particular reason that the pad_with_graphs function uses the numpy library rather than the jax.numpy library. It looks like every numpy function in there can just ...
  • GrantMcConachie
  • 1
  • Opened 
    on Jul 5, 2024
  • #52

google-deepmind/jraph
posting flax examples?

Hi, I ve used jraph layers in Flax. Is there an examples section that I can contribute to? GAT in JAX/Flax: https://github.com/mashaan14/YouTube-channel/blob/main/notebooks/2024_03_18_jraph_GAT.ipynb ...
  • mashaan14
  • Opened 
    on Mar 21, 2024
  • #51

google-deepmind/jraph
Jraph Flax incorporation

Hi! I saw on the Haiku repo that they are urging people to move to flax, and was just wondering if jraph has any intention of moving to a flax framework? Thanks!
  • GrantMcConachie
  • 2
  • Opened 
    on Nov 22, 2023
  • #49

google-deepmind/jraph
How to create custom batches?

Assume I have multiple graphs (nodes, senders, receivers and graph label). How would one create batches that would work with jax.vmap?
  • MichaelMMeskhi
  • Opened 
    on Jul 17, 2023
  • #48

google-deepmind/jraph
pad_with_graph using numpy instead of jax.numpy

I just found this issue while using this util function in my project, as shown here: https://github.com/deepmind/jraph/blob/51f5990104f7374492f8f3ea1cbc47feb411c69c/jraph/_src/utils.py#L614-L619 Is there ...
  • ziyuanzhao2000
  • Opened 
    on Mar 14, 2023
  • #46

google-deepmind/jraph
[Potential Bug] Senders and Receivers can't be None as specified in docstrings

Hello, I think there might be a bug in the node aggregation fn when using GCN. The docstrings state that we can set the senders and receivers to None for graphs that have no edges but due to using the ...
  • EdanToledo
  • Opened 
    on Jan 24, 2023
  • #45

google-deepmind/jraph
Examples of regression task on the nodes features

Hi, Could you provide an example on how to perform regression prediction on nodes features? Thank you!
  • nbruciaferri
  • 1
  • Opened 
    on Sep 30, 2022
  • #43

google-deepmind/jraph
Examples for OGB Node/Link Pred Datasets

Hi, Do you have any examples for ogb node/link prediction datasets? If not, can you provide some examples? Thanks!
  • kaansancak
  • Opened 
    on Sep 26, 2022
  • #42
Issue origami icon

Learn how you can use GitHub Issues to plan and track your work.

Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub Issues
ProTip! 
Press the
/
key to activate the search input again and adjust your query.
Issue origami icon

Learn how you can use GitHub Issues to plan and track your work.

Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub Issues
ProTip! 
Restrict your search to the title by using the in:title qualifier.
Issue search results · GitHub