issues
search
patrick-kidger
/
quax
Multiple dispatch over abstract array types in JAX.
Apache License 2.0
100
stars
2
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
`scan` primitive implementation missing
#29
colehaus
opened
1 week ago
4
LoRA that doesn't require memory for zero gradients of the underlying matrices
#28
colehaus
opened
2 weeks ago
6
Updated ecosystem docs to match style. Added Quaxed & unxt.
#27
nstarman
closed
3 weeks ago
1
Updated ecosystem docs to match style. Added Quaxed.
#26
patrick-kidger
opened
1 month ago
4
Added sharding parameter: fix for JAX 0.4.31
#25
patrick-kidger
closed
1 month ago
0
feat: pass precedence to plum dispatcher
#24
nstarman
closed
1 month ago
2
feat: add type annotations
#23
nstarman
closed
1 month ago
6
v004
#22
patrick-kidger
closed
2 months ago
0
cut new release
#21
nstarman
closed
1 month ago
2
feat: add py.typed marker
#20
nstarman
closed
2 months ago
1
`py.typed`
#19
nstarman
closed
2 months ago
1
bump: pre-commit
#18
nstarman
closed
3 months ago
0
feat: prng split works on jax keys
#17
nstarman
closed
2 months ago
5
Implemented jax.lax.while primitive
#16
ymahlau
closed
1 month ago
7
How to implement jax.lax.while with quax
#15
ymahlau
closed
1 month ago
2
feat: convenience construction of a threefry
#14
nstarman
closed
2 months ago
7
quax boundary recursive wrapping of a pytree
#13
nstarman
closed
5 months ago
2
Compatibility with `jaxtyping`
#12
nstarman
closed
2 months ago
2
jit and vamp best practices
#11
nstarman
opened
7 months ago
1
Interest in add array-api module
#10
nstarman
opened
7 months ago
15
g->kg
#9
patrick-kidger
closed
7 months ago
0
Unitful example
#8
krzysztofrusek
closed
7 months ago
1
maybe rewrite
#7
patrick-kidger
closed
7 months ago
0
add `__pow__`
#6
nstarman
closed
8 months ago
3
`quaxify` on a `jax.grad`
#5
nstarman
closed
7 months ago
10
No longer using a dynamic trace.
#4
patrick-kidger
opened
8 months ago
0
jnp.zeros{,_like} no longer produces a Zero.
#3
patrick-kidger
opened
8 months ago
2
Support for `zeros_like` and related
#2
nstarman
closed
7 months ago
9
name rule for debugging purposes
#1
nstarman
closed
9 months ago
1