krzysztofrusek

krzysztofrusek's Stars

• google-research/google-research

  Google Research

  Language:Jupyter Notebook34.6k7551.3k8k

• google-deepmind/alphafold

  Open source code for AlphaFold 2.

  Language:Python13.1k2288702.3k

• google-research/scenic

  Scenic: A Jax Library for Computer Vision Research and Beyond

  Language:Python3.4k40268440

• google-research/t5x

  Language:Python2.7k36141309

• google-research/big_vision

  Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

  Language:Jupyter Notebook2.5k3961163

• google/brax

  Massively parallel rigidbody physics simulation on accelerator hardware.

  Language:Jupyter Notebook2.5k33362265

• google-deepmind/mctx

  Monte Carlo tree search in JAX

  Language:Python2.4k2852195

• google/neural-tangents

  Fast and Easy Infinite Neural Networks in Python

  Language:Jupyter Notebook2.3k62156227

• google-deepmind/optax

  Optax is a gradient processing and optimization library for JAX.

  Language:Python1.8k35266199

• patrick-kidger/diffrax

  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

  Language:Python1.5k23359137

• patrick-kidger/jaxtyping

  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

  Language:Python1.3k1314766

• google/jaxopt

  Hardware accelerated, batchable and differentiable optimizers in JAX.

  Language:Python9421820969

• google/evojax

  Language:Jupyter Notebook861233192

• google-research/jax3d

  Language:Python740285696

• google-deepmind/distrax

  Language:Python542184633

• google-deepmind/dm_pix

  PIX is an image processing library in JAX, for JAX.

  Language:Python39710926

• patrick-kidger/sympy2jax

  Turn SymPy expressions into trainable JAX expressions.

  Language:Python3269413

• google-deepmind/enn

  Language:Python299141060

• google-deepmind/kfac-jax

  Second Order Optimization and Curvature Estimation with K-FAC in JAX.

  Language:Python255111924

• jax-ml/oryx

  Oryx is a library for probabilistic programming and deep learning built on top of Jax.

  Language:Python236102011

• google-research/ott

  Language:Python213151118

• google-deepmind/neural_networks_chomsky_hierarchy

  Neural Networks and the Chomsky Hierarchy

  Language:Python19112820

• google-deepmind/neural_testbed

  Language:Jupyter Notebook19112418

• google-research/graphworld

  Language:Python1739430

• google/network-opt

  A library for topological network optimization

  Language:C++1518317

• google-deepmind/tf2jax

  Language:Python1097612

• google-deepmind/dm_aux

  Language:Python63837

• google/codex

  Data compression in JAX

  Language:Jupyter Notebook59436

• iree-org/iree-jax

  Language:Jupyter Notebook50341719

• nekeal/fige

  ML project for geometric figures recognizing

  Language:Python210