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