This library provides some utility function and useful layers that extend the Equinox library.
The aim was to create them to be the PyTorch equivalent, i.e. to match their PyTorch counterpart's output.
-
Notifications
You must be signed in to change notification settings - Fork 0
Artur-Galstyan/jaxonlayers
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Additional layers for use with Equinox and JAX
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published