Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
cgarciae authored Apr 19, 2023
1 parent 22848ea commit 9ee3fe6
Showing 1 changed file with 0 additions and 12 deletions.
12 changes: 0 additions & 12 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,18 +12,6 @@ A _dead simple_ Python package for creating custom JAX pytree objects.
* Its compatible with both `dataclasses` and regular classes
* It has no intention of supporting Neural Network use cases (e.g. partitioning)

<details><summary>What about Equinox, Treeo, etc?</summary>

Most pytree-based neural network libraries start simple but end up adding
a lot of features that are not needed for simple pytree objects. `flax.struct.PytreeNode`
is the simplest one out there, but it has two downsides:

1. Forces you to use `dataclasses`, which is not a bad thing but not always
what you want.
2. It requires you to install `flax` just to use it.

</details>

## Installation

```bash
Expand Down

0 comments on commit 9ee3fe6

Please sign in to comment.