Skip to content

Activity

Showing most recent first

[Mosaic] Add relayout for (1, 128 * packing) -> (packing, 128).

Micky774pushed 28 commits to main • c577087…8b95853 • 
on May 28, 2024

Add support for device kwarg in astype, and add matching utility func

Force push
Micky774force pushed to astype-device • 33d2df7…7c7c3b8 • 
on May 24, 2024

Enable a lot of previously disabled tests now that compute offload wo…

Micky774pushed 82 commits to main • d33a568…c577087 • 
on May 24, 2024

Finalize jax.numpy array API compliance

Force push
Micky774force pushed to jnp-array-api • 10c5c2d…cdfb270 • 
on May 21, 2024

Finalize jax.numpy array API compliance

Force push
Micky774force pushed to jnp-array-api • fd88e2a…10c5c2d • 
on May 21, 2024

Finalize jax.numpy array API compliance

Force push
Micky774force pushed to jnp-array-api • 4201eaf…fd88e2a • 
on May 21, 2024

Refactor & test internal deprecation APIs

Micky774pushed 38 commits to main • 210f8bb…d33a568 • 
on May 21, 2024

Finalize jax.numpy array API compliance

Force push
Micky774force pushed to jnp-array-api • bb5898f…4201eaf • 
on May 21, 2024

WIP jax.numpy array API compliance finalization

Micky774created jnp-array-api • bb5898f • 
on May 21, 2024

Deleted branch

Micky774deleted linalg-depr • 
on May 18, 2024

Use absl.flags.FlagHolder to defined --mosaic_gpu_debug

Micky774pushed 3 commits to main • 5e2710c…210f8bb • 
on May 17, 2024

Add support for device kwarg in astype

Force push
Micky774force pushed to astype-device • e3fdebb…33d2df7 • 
on May 17, 2024

Merge pull request jax-ml#21261 from superbobry:mypy-ruff

Micky774pushed 47 commits to main • 0501d3d…5e2710c • 
on May 17, 2024

Rename rcond/tol to rtol in linalg.matrix_rank and linalg.pinv

Force push
Micky774force pushed to linalg-depr • 9c06888…5cc255b • 
on May 14, 2024

Rename rcond/tol to rtol in linalg.matrix_rank and linalg.pinv

Force push
Micky774force pushed to linalg-depr • 76379cb…9c06888 • 
on May 14, 2024

Rename rcond/tol to rtol in linalg.matrix_rank and linalg.pinv

Force push
Micky774force pushed to linalg-depr • 1c3a905…76379cb • 
on May 14, 2024

Rename rcond/tol to rtol in linalg.matrix_rank and linalg.pinv

Force push
Micky774force pushed to linalg-depr • 87fbea1…1c3a905 • 
on May 14, 2024

Merge pull request jax-ml#21223 from jakevdp:kl-div-doc

Micky774pushed 2 commits to main • bb5787d…0501d3d • 
on May 14, 2024

Finalize deprecations of several APIs

Micky774pushed 59 commits to main • 13a1955…bb5787d • 
on May 14, 2024

Add support for device kwarg in astype

Force push
Micky774force pushed to astype-device • 60eaabc…e3fdebb • 
on May 14, 2024

Add support for device kwarg in astype

Force push
Micky774force pushed to astype-device • f0f52a5…60eaabc • 
on May 10, 2024

Merge pull request jax-ml#21167 from jakevdp:einsum-path-func

Micky774pushed 13 commits to main • 1cb6971…13a1955 • 
on May 10, 2024

Add support for device kwarg in astype

Force push
Micky774force pushed to astype-device • 5445230…f0f52a5 • 
on May 10, 2024

Add support for device kwarg in astype

Force push
Micky774force pushed to astype-device • 593f01c…5445230 • 
on May 9, 2024

Add support for device kwarg in astype

Force push
Micky774force pushed to astype-device • 72958d1…593f01c • 
on May 9, 2024

Add support for device kwarg in astype

Force push
Micky774force pushed to astype-device • 22b71b8…72958d1 • 
on May 9, 2024

Deleted branch

Micky774deleted reshape • 
on May 9, 2024

Deprecate newshape argument of jnp.reshape

Force push
Micky774force pushed to reshape • 708df0f…79005c1 • 
on May 9, 2024

Merge pull request jax-ml#21107 from justinjfu/pallas_splash_test_fix

Micky774pushed 33 commits to main • 11da3df…1cb6971 • 
on May 9, 2024

Deprecate newshape argument of jnp.reshape

Force push
Micky774force pushed to reshape • 45039bc…708df0f • 
on May 9, 2024