Metadata-Version: 2.4
Name: jaxonlayers
Version: 0.2.1
Summary: Additional layers and functions that extend Equinox
Requires-Python: >=3.13
Requires-Dist: beartype>=0.21.0
Requires-Dist: equinox>=0.13.0
Requires-Dist: jax>=0.8.0
Requires-Dist: jaxtyping>=0.3.2
Description-Content-Type: text/markdown

# jaxonlayers

This library provides some utility function and useful layers that extend the [Equinox](https://github.com/patrick-kidger/equinox) library.

The aim was to create them to be the PyTorch equivalent, i.e. to match their PyTorch counterpart's output.
