Skip to content

Commit 8487301

Browse files
Update the WIP section.
1 parent a6fba20 commit 8487301

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,10 @@ lora_linear = lora.loraify(linear, rank=2, key=key3)
7171

7272
Right now, the following are not supported:
7373

74-
- Control flow primitives (e.g. `jax.lax.cond`).
74+
- `jax.lax.scan_p`
7575
- `jax.custom_vjp`
7676

77-
It should be fairly straightforward to add support for these; open an issue or pull request.
77+
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`. :) )
7878

7979
## See also: other libraries in the JAX ecosystem
8080

0 commit comments

Comments
 (0)