-
-
-
stablehlo Public
Forked from openxla/stablehloBackward compatible ML compute opset inspired by HLO/MHLO
MLIR Apache License 2.0 UpdatedSep 5, 2024 -
jax-triton Public
Forked from jax-ml/jax-tritonjax-triton contains integrations between JAX and OpenAI Triton
-
superlu Public
Forked from xiaoyeli/superluSupernodal sparse direct solver. https://portal.nersc.gov/project/sparse/superlu/
C Other UpdatedJul 25, 2024 -
benchmark Public
Forked from google/benchmarkA microbenchmark support library
-
equinox Public
Forked from patrick-kidger/equinoxElegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Python Apache License 2.0 UpdatedJun 10, 2024 -
nanobind Public
Forked from wjakob/nanobindnanobind — Seamless operability between C++17 and Python
C++ BSD 3-Clause "New" or "Revised" License UpdatedMay 23, 2024 -
nccl Public
Forked from NVIDIA/ncclOptimized primitives for collective multi-GPU communication
C++ Other UpdatedJan 9, 2024 -
-
llvm-project Public
Forked from llvm/llvm-projectThe LLVM Project is a collection of modular and reusable compiler and toolchain technologies.
-
bloaty Public
Forked from google/bloatyBloaty: a size profiler for binaries
C++ Apache License 2.0 UpdatedDec 1, 2023 -
flatbuffers Public
Forked from google/flatbuffersFlatBuffers: Memory Efficient Serialization Library
C++ Apache License 2.0 UpdatedNov 18, 2023 -
numpy Public
Forked from numpy/numpyThe fundamental package for scientific computing with Python.
C BSD 3-Clause "New" or "Revised" License UpdatedNov 13, 2023 -
gloo Public
Forked from facebookincubator/glooCollective communications library with various primitives for multi-machine training.
C++ Other UpdatedOct 26, 2023 -
xla Public
Forked from openxla/xlaA machine learning compiler for GPUs, CPUs, and ML accelerators
C++ Apache License 2.0 UpdatedSep 23, 2023 -
triton Public
Forked from triton-lang/tritonDevelopment repository for the Triton language and compiler
C++ MIT License UpdatedAug 9, 2023 -
tensorflow Public
Forked from tensorflow/tensorflowComputation using data flow graphs for scalable machine learning
C++ Apache License 2.0 UpdatedFeb 13, 2023 -
-
jaxlib-feedstock Public
Forked from conda-forge/jaxlib-feedstockA conda-smithy repository for jaxlib.
Shell BSD 3-Clause "New" or "Revised" License UpdatedAug 15, 2022 -
pandas Public
Forked from pandas-dev/pandasFlexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more
Python BSD 3-Clause "New" or "Revised" License UpdatedAug 3, 2022 -
numexpr Public
Forked from pydata/numexprFast numerical array expression evaluator for Python, NumPy, PyTables, pandas, bcolz and more
Python MIT License UpdatedJul 19, 2022 -
-
diffrax Public
Forked from patrick-kidger/diffraxNumerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.
Python Apache License 2.0 UpdatedApr 6, 2022 -
pybind11 Public
Forked from pybind/pybind11Seamless operability between C++11 and Python
-
jaxlib-gha-minimal-failing-example Public
Forked from matthewfeickert/jaxlib-gha-minimal-failing-exampleDemonstration of pyhf Issue 1501
Python MIT License UpdatedJul 7, 2021 -
OpenBLAS Public
Forked from OpenMathLib/OpenBLASOpenBLAS is an optimized BLAS library based on GotoBLAS2 1.13 BSD version.
Fortran BSD 3-Clause "New" or "Revised" License UpdatedFeb 18, 2021 -
numpyro Public
Forked from pyro-ppl/numpyroProbabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Python Apache License 2.0 UpdatedMay 8, 2020 -
opt_einsum Public
Forked from dgasmith/opt_einsum⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Python MIT License UpdatedApr 15, 2020 -
flax Public
Forked from google/flaxFlax is a neural network library for JAX that is designed for flexibility.
1 UpdatedMar 20, 2020