Neil Girdhar 1e580457ba Repair various type errors

2024-03-13 15:13:56 -04:00

..

__init__.py

[Pallas] Add async_copy_to and async_remote_copy_to for doing DMAs.

2023-09-14 14:32:08 -07:00

discharge.py

[mutable-arrays] allow state effects in jit by building in run_state

2024-02-29 21:50:19 -08:00

indexing.py

Slice.from_slice now works for slices with a negative start index

2024-01-29 13:13:06 +00:00

primitives.py

[mutable-arrays] move MutableArray, add eager, improve tests, fix bug

2024-03-01 15:03:23 -08:00

types.py

Repair various type errors

2024-03-13 15:13:56 -04:00

utils.py

integrate attrs in jax.jit

2024-01-27 17:44:43 -08:00