krzysztofrusek's Stars
jakevdp/PythonDataScienceHandbook
Python Data Science Handbook: full text in Jupyter Notebooks
astral-sh/uv
An extremely fast Python package and project manager, written in Rust.
CamDavidsonPilon/Probabilistic-Programming-and-Bayesian-Methods-for-Hackers
aka "Bayesian Methods for Hackers": An introduction to Bayesian methods + probabilistic programming with a computation/understanding-first, mathematics-second point of view. All in pure Python ;)
modularml/mojo
The Mojo Programming Language
google-deepmind/graphcast
facebookresearch/nevergrad
A Python toolbox for performing gradient-free optimization
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
google-deepmind/penzai
A JAX research toolkit for building, editing, and visualizing neural networks.
google-deepmind/materials_discovery
google-deepmind/chex
google/temporian
Temporian is an open-source Python library for preprocessing ⚡ and feature engineering 🛠 temporal data 📈 for machine learning applications 🤖
google/paxml
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
srush/Autodiff-Puzzles
patrick-kidger/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
google-deepmind/treescope
An interactive HTML pretty-printer for machine learning research in IPython notebooks.
patrick-kidger/optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
normal-computing/posteriors
Uncertainty quantification with PyTorch
google-deepmind/synjax
endia-org/Endia
Build and train Neural Networks in Mojo
jax-ml/bayeux
State of the art inference for your bayesian models.
srush/do-we-need-attention
google-deepmind/digraph_transformer
patrick-kidger/quax
Multiple dispatch over abstract array types in JAX.
jax-ml/jax-ai-stack
itsdaniele/jeometric
Graph neural networks in JAX.
JiaYaobo/fenbux
A Simple Statistical Distribution Library in JAX
google-deepmind/ccbo
This repo contains the code associated to the paper: "Constrained Causal Bayesian Optimization" by Aglietti Virginia, Alan Malek, Ira Ktena, and Silvia Chiappa. International Conference on Machine Learning. PMLR, 2023.
devvrit/SONew
BNN-UPC/Atom_Neural_Traffic_Compression
This repository contains de code and instructions to train the models and prepare the datasets for the experiments in the paper "Atom: Neural Traffic Compression with Spatio-Temporal Graph Neural Networks" accepted at the 2nd ACM CONEXT GNNet 2023 Workshop.
gsd-authors/gsd
Reference implementation of generalised score distribution in python