Foldseek enables fast and sensitive comparisons of large structure sets.

🌊 Julia software for fast, friendly, flexible, ocean-flavored fluid dynamics on CPUs and GPUs

A platform for managing machine learning experiments

Friends don't let friends make certain types of data visualization - What are they and why are they bad.

Multiple dispatch in Python

Compositional Linear Algebra

diffusionjax is a simple and accessible diffusion models package in JAX

PDEBench: An Extensive Benchmark for Scientific Machine Learning

Computational Fluid Dynamics in JAX

Code used by the "Clifford Group Equivariant Neural Networks" paper.

Play atmospheric modelling like it's LEGO.

Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other scientific publications.

A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.

Fast and Easy Infinite Neural Networks in Python

Free monospaced font with programming ligatures

A functional training loops library for JAX

An OpenStreetMap-based project for creating a map of the world's railway infrastructure.

PyTorch implementation of normalizing flow models

Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.

Protein Graph Library

A framework for composing Neural Processes in Python

E(3) Steerable Graph Neural Network

Geometric kernels on manifolds, meshes and graphs

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

Training neural network potentials

A Graph Neural Network Library in Jax

A toy diffusion with JAX

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/