Morph MORPH
SIGN IN SIGN UP
dhruvhead / jax UNCLAIMED
0 0 0 Python

ACTIVITY

Commits Reviews
week month year
COMMITS
20
in the last week
CONTRIBUTORS
7
active
STARS
0
total
FORKS
0
total
TOP CONTRIBUTORS
J
jax authors
7 commits
P
Peter Hawkins
4 commits
Y
Yash Katariya
3 commits
C
Cristian Garcia
3 commits
D
Dirk Hornung
1 commit
R
Robert Dyro
1 commit
M
Matthew Johnson
1 commit
RECENT COMMITS
J
Update XLA dependency to use revision http://github.com/openxla/xla/commit/496bd4bd49db9ecbffd85da630b49c860b724604
jax authors
5832e86
Y
Use program_order_p even if exclude_argnames is None
Yash Katariya
8a92dd9
J
Update XLA dependency to use revision http://github.com/openxla/xla/commit/69325769ed5c4191b16e7f5fd37d5d5ea114a18d
jax authors
eabf536
D
Reverts 2610691a3fcfc16d5fd09e714f492f518978622c
Dirk Hornung
a2f2d71
C
Explicitly propagate and specify strip_memory_space in state discharge callers.
Cristian Garcia
fefdef2
C
Support mutable operations (`set`, `swap`, `addupdate`) on `ReshapeTransform` and `BitcastTransform` views during state discharge.
Cristian Garcia
fd3802e
R
[pallas:fuser] Support full-dimension (dynamic_)slice and (dynamic_)update_slice
Robert Dyro
05fcbe4
J
Merge pull request #40306 from mattjj:push-unytxtzpuxsr
jax authors
264952c
Y
Add `exclude_argnames` on program_order.
Yash Katariya
94b575c
J
Merge pull request #40305 from hawkinsp:py315
jax authors
d551c34
M
[docs] more tweaks to complex number description
Matthew Johnson
c6bfad7
P
Run wheel tests using the git commit the wheels were built from.
Peter Hawkins
b2e4a13
J
Merge pull request #40231 from VaggelisGian:fix-intersect-setxor-size0
jax authors
b94bd6c
P
Use a rules_python manifest instead of patching rules_python for 3.15.
Peter Hawkins
75b9bd2
J
Merge pull request #40295 from hawkinsp:py315
jax authors
01293a5
C
Move with_memory_space_constraint primitive to jax._src.state.primitives.
Cristian Garcia
ab1e448
J
Merge pull request #40299 from hawkinsp:deps
jax authors
00894c5
Y
Dedup prev_outs with cur_inps so we don't barrier on the same input twice. In other words, if some prev_outs are in cur_inps, then skip those prev_outs from opt_barrier (since they are already covered by cur_inps).
Yash Katariya
ae115c2
P
Drain garbage collector before reference cycle tests.
Peter Hawkins
1214954
P
[Pallas/GPU interpreter] Relax test tolerances and use NumPy matmul for reference.
Peter Hawkins
b277e8e