Skip to content

Commit

Permalink
Update the WIP section.
Browse files Browse the repository at this point in the history
  • Loading branch information
patrick-kidger authored Nov 11, 2024
1 parent a6fba20 commit 8487301
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -71,10 +71,10 @@ lora_linear = lora.loraify(linear, rank=2, key=key3)

Right now, the following are not supported:

- Control flow primitives (e.g. `jax.lax.cond`).
- `jax.lax.scan_p`
- `jax.custom_vjp`

It should be fairly straightforward to add support for these; open an issue or pull request.
It should be fairly straightforward to add support for these; open an issue or pull request. (We've already got `jax.custom_jvp`, `jax.lax.cond_p`, and `jax.lax.while_p`. :) )

## See also: other libraries in the JAX ecosystem

Expand Down

0 comments on commit 8487301

Please sign in to comment.