This repository was archived by the owner on May 21, 2025. It is now read-only.

This repository was archived by the owner on May 21, 2025. It is now read-only.

How to create custom batches? #48

Open

opened

on Jul 17, 2023

Assume I have multiple graphs (nodes, senders, receivers and graph label). How would one create batches that would work with jax.vmap?

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Type

No type

Fields

No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests