23755 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
srayasam-amd	40a993797a	Update ci_build	2025-02-13 18:46:06 +05:30
srayasam-amd	ae9f449c13	Update ci_build	2025-02-13 17:15:13 +05:30
srayasam-amd	c95d267620	Update build_wheels.py	2025-02-13 16:06:56 +05:30
srayasam-amd	90f95a2e57	Update ci_build	2025-02-13 15:41:25 +05:30
srayasam-amd	2aa8e23b93	Update Dockerfile.manylinux_2_28_x86_64.rocm	2025-02-13 11:17:53 +05:30
srayasam-amd	96315972a8	Update ci_build.sh	2025-02-13 11:06:55 +05:30
srayasam-amd	082d05757f	Update ci_build.sh	2025-02-13 11:01:03 +05:30
srayasam-amd	d3cdd4f72d	Update ci_build.sh	2025-02-13 10:56:45 +05:30
srayasam-amd	6f8b984b89	Update Dockerfile.manylinux_2_28_x86_64.rocm	2025-02-13 10:51:17 +05:30
srayasam-amd	3e64d0b627	Update ci_build.sh	2025-02-13 10:44:44 +05:30
srayasam-amd	6d50f2a2c2	Update ci_build.sh	2025-02-13 10:30:47 +05:30
srayasam-amd	bdea57c455	Update Dockerfile.manylinux_2_28_x86_64.rocm	2025-02-13 10:27:00 +05:30
srayasam-amd	d917a22513	Update Dockerfile.manylinux_2_28_x86_64.rocm	2025-02-12 21:07:02 +05:30
srayasam-amd	3ba06823e8	Update Dockerfile.manylinux_2_28_x86_64.rocm	2025-02-12 15:39:06 +05:30
srayasam-amd	58f2c0e44d	Update ci_build.sh	2025-02-12 14:52:10 +05:30
srayasam-amd	3403473e91	Update ci_build.sh	2025-02-12 13:46:14 +05:30
srayasam-amd	5001365663	Update ci_build.sh	2025-02-12 11:56:00 +05:30
srayasam-amd	a53822fa7c	Update ci_build.sh ... adding if condition for gpu_Device_targets	2025-02-12 11:53:57 +05:30
srayasam-amd	5da1c4e8df	Update ci_build	2025-02-10 18:08:40 +05:30
srayasam-amd	460b3c357d	Update build_wheels.py	2025-02-10 18:02:52 +05:30
srayasam-amd	f72eb95f6d	Update ci_build.sh	2025-02-10 18:01:30 +05:30
srayasam-amd	a3d3481867	Update Dockerfile.manylinux_2_28_x86_64.rocm ... adding gpu_device_targets	2025-02-10 17:56:51 +05:30
Ruturaj Vaidya	39f7db687e	Change jaxlib/rocm/BUILD nanobind_extension to pybind (#226)	2025-02-07 16:07:02 -06:00
Ruturaj Vaidya	dceb5310fe	[ROCm] Implement RNN support (#217)	2025-02-07 11:08:12 -06:00
Ruturaj Vaidya	da40f36d00	Fix run_multi_gpu script multi-gpu issue and refactor code (#208)	2025-01-19 18:03:00 -06:00
Ruturaj Vaidya	f99fbcb3c5	Add gfx12xx archs (#205)	2025-01-14 10:52:00 -06:00
Kiran Thumma	5d56aaf1e3	Merge pull request #201 from ROCm/rocm-jaxlib-v0.4.35-qa-fix2 ... buildkit option for docker build	2025-01-13 06:43:44 -06:00
kithumma	f075f904b2	buildkit option for docker build	2025-01-11 06:09:27 +00:00
Kiran Thumma	c5e32ac558	Merge pull request #195 from ROCm/rocm-jaxlib-v0.4.35-qa-fix ... fix cdn	2025-01-08 06:12:51 -06:00
kithumma	590ac67c4b	fix cdn	2024-12-30 17:11:51 +00:00
Ruturaj Vaidya	8985f20e4f	Merge pull request #169 from ROCm/ci_qa_sync ... [JAX] Ci qa sync	2024-12-02 10:28:21 -06:00
Ruturaj Vaidya	fab53b06f9	[ROCm] Update XLA commit hash (#168)	2024-12-02 06:51:34 -06:00
Ruturaj Vaidya	20362f83c6	[ROCm] Remove exit on error (#167)	2024-12-02 06:51:25 -06:00
Ruturaj4	010aea0c56	Add argument to override base docker in dockerfile	2024-12-02 06:49:40 -06:00
Ruturaj4	3b3083bbc4	[ROCm] ci build and dockerfile changes	2024-12-02 06:49:32 -06:00
Ruturaj4	b8d2dedeed	[ROCm] Change run_multi_gpu set opts	2024-12-02 06:49:22 -06:00
Ruturaj4	a5a1417587	BUILD file fixes and linter run	2024-12-02 06:49:15 -06:00
Ruturaj4	addca5d517	[ROCm] fix jax and wheelhouse relative paths	2024-12-02 06:49:10 -06:00
Ruturaj4	39c0450638	[ROCm] ci build fixes	2024-12-02 06:49:01 -06:00
Peter Hawkins	81991d87c8	JAX release 0.4.35	2024-10-22 15:00:23 -04:00
Peter Hawkins	e4f3f8f064	Use libtpu releases rather than libtpu-nightly for jax[tpu]. ... PiperOrigin-RevId: 688632409	2024-10-22 11:47:07 -07:00
jax authors	1c6b0a9193	Merge pull request #24465 from jakevdp:fix-mypy ... PiperOrigin-RevId: 688632024	2024-10-22 11:45:27 -07:00
jax authors	9a2dd19a92	Merge pull request #21524 from andportnoy:aportnoy/unknown-platform-lowering-warning ... PiperOrigin-RevId: 688630259	2024-10-22 11:40:39 -07:00
jax authors	1e41d5ef6f	Merge pull request #24452 from jakevdp:insert-doc ... PiperOrigin-RevId: 688624762	2024-10-22 11:26:38 -07:00
jax authors	1a2737b72b	Merge pull request #24467 from andportnoy:patch-2 ... PiperOrigin-RevId: 688620752	2024-10-22 11:17:15 -07:00
Jake VanderPlas	849850216d	fix mypy error	2024-10-22 11:10:10 -07:00
jax authors	92aa9a3990	Merge pull request #24463 from jakevdp:ml-dtypes-version ... PiperOrigin-RevId: 688617061	2024-10-22 11:07:00 -07:00
Ayaka	c60bafcc33	[Pallas TPU] Fix lowering for jnp.remainder ... Fixes https://github.com/jax-ml/jax/issues/24027 PiperOrigin-RevId: 688614799	2024-10-22 11:01:58 -07:00
Andrey Portnoy	637898493e	Add back the import of jtu in flash_attention.py ... This was erroneously removed in de3191fab.	2024-10-22 13:37:35 -04:00
Andrey Portnoy	2aaa108f06	Raise an error when registering a lowering for an unknown platform	2024-10-22 13:29:48 -04:00