krzysztofrusek's Stars
karpathy/nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
typst/typst
A new markup-based typesetting system that is powerful and easy to learn.
google-research/tuning_playbook
A playbook for systematically maximizing the performance of deep learning models.
google/dopamine
Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
google/latexify_py
A library to generate LaTeX expression from Python code.
qjcg/awesome-typst
Awesome Typst Links
patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
jax-md/jax-md
Differentiable, Hardware Accelerated, Molecular Dynamics
aesara-devs/aesara
Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.
blackjax-devs/blackjax
BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
probml/dynamax
State Space Models library in JAX
ott-jax/ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
google/autobound
AutoBound automatically computes upper and lower bounds on functions.
matthewwardrop/formulaic
A high-performance implementation of Wilkinson formulas for Python.
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
jax-ml/oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
google-deepmind/neural_networks_chomsky_hierarchy
Neural Networks and the Chomsky Hierarchy
probml/sts-jax
Structural Time Series in JAX
google-deepmind/PGMax
Loopy belief propagation for factor graphs on discrete variables in JAX
openxla/community
Stores documents and resources used by the OpenXLA developer community
google-research/spherical-cnn
patrick-kidger/typst_pyimage
Typst extension, adding support for generating figures using inline Python code
darrenjw/fp-ssc-course
An introduction to functional programming for scalable statistical computing
blackjax-devs/sampling-book
Tutorials and sampling algorithm comparisons
ASEM000/kernex
Stencil computations in JAX
m-wojnar/reinforced-lib
Reinforcement learning library
eserie/wax-ml
A Python library for machine-learning and feedback loops on streaming data
probml/rebayes
Recursive Bayesian Estimation (Sequential / Online Inference)
google-research/dataclass_array
Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
jax-ml/coix
Inference Combinators in JAX