-
jax Public
Forked from jax-ml/jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
-
jax-ai-stack Public
Forked from jax-ml/jax-ai-stack -
coix Public
Forked from jax-ml/coixInference Combinators in JAX
Jupyter Notebook Apache License 2.0 UpdatedMay 17, 2025 -
jax-llm-examples Public
Forked from jax-ml/jax-llm-examplesPython Apache License 2.0 UpdatedMay 8, 2025 -
numpyro Public
Forked from pyro-ppl/numpyroProbabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
-
jaxopt Public
Forked from google/jaxoptHardware accelerated, batchable and differentiable optimizers in JAX.
Python Apache License 2.0 UpdatedApr 9, 2025 -
flax Public
Forked from google/flaxFlax is a neural network library for JAX that is designed for flexibility.
-
scikit-learn Public
Forked from scikit-learn/scikit-learnscikit-learn: machine learning in Python
-
numpy Public
Forked from numpy/numpyNumpy main repository
-
optax Public
Forked from google-deepmind/optaxOptax is a gradient processing and optimization library for JAX.
Python Apache License 2.0 UpdatedJan 15, 2025 -
diffrax Public
Forked from patrick-kidger/diffraxNumerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
-
keras Public
Forked from keras-team/kerasDeep Learning for humans
-
array-api-tests Public
Forked from data-apis/array-api-testsTest suite for the PyData Array APIs standard
-
orbax Public
Forked from google/orbaxOrbax provides common utility libraries for JAX users.
Python Apache License 2.0 UpdatedNov 20, 2024 -
array-api-compat Public
Forked from data-apis/array-api-compatCompatibility layer for NumPy to support the Array API
-
PythonDataScienceHandbook Public
Python Data Science Handbook: full text in Jupyter Notebooks
-
-
equinox Public
Forked from patrick-kidger/equinoxElegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
-
-
-
-
sphinx-autodoc-typehints Public
Forked from tox-dev/sphinx-autodoc-typehintsType hints support for the Sphinx autodoc extension
Python Other UpdatedApr 22, 2024 -
jakevdp.github.io-source Public
Source for my Pythonic Perambulations blog
-
WhirlwindTourOfPython Public
The Jupyter Notebooks behind my OReilly report, "A Whirlwind Tour of Python"
-
maxtext Public
Forked from AI-Hypercomputer/maxtextA simple, performant and scalable Jax LLM!
-
diffusers Public
Forked from huggingface/diffusers🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
-
-
dynamax Public
Forked from probml/dynamaxState Space Models library in JAX
-
supersmoother Public
Efficient pure Python implementation of Friedman's Supersmoother
{{ message }}