LoreonLabsPlatform

Overview

Intelligence

Markets
Builders
Research
Narratives
Ecosystems
Launchpads

Discover

Search
Sources

Python

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

PythonEmerging

Stars

—

Forks

—

Contributors

8

Last push

5mo ago

Recent commits

Latest commits.

Reverts 4a5ad80ec771ec08bea2b28169d2f298aecbb46e
5c4c7f0Peter Hawkins5mo ago
Merge pull request #34682 from olupton:workaround-tcgen05-llvm-errors
8288263jax authors5mo ago
[Mosaic GPU] Clean up all the linter warnings in `dialect_lowering.py`.
40c3359Benjamin Chetioui5mo ago
[Pallas/Mosaic GPU] Remove unnecessary jaxlib version guard-related code.
a5854e4Benjamin Chetioui5mo ago
Update XLA dependency to use revision http://github.com/openxla/xla/commit/80529bdfc1c3bc206d0809a474031e9c9008c9f0
1066aa7jax authors

5mo ago

[mosaic] Added a macro which defines AbslStringify for all MLIR enums in the enclosing namespace

150a6d9Sergei Lebedev5mo ago

[mosaic] Avoid using the deprecated `output_memory_colors` in `CustomCallBackendConfig`

79fd127Sergei Lebedev5mo ago

Regenerate python stubs for nanobind 2.11.0.

6fb3f3dPeter Hawkins5mo ago

Top contributors

Builders behind this project.

Google-ML-Automation