A deep latent generative model for learning biosystems

Code for the paper: "Abide by the Law and Follow the Flow: Conservation Laws for Gradient Flows".

Optax is a gradient processing and optimization library for JAX.

Flax is a neural network library for JAX that is designed for flexibility.

Orbax provides common checkpointing and persistence utilities for JAX users

Repository to hold implementations for Joint optimization of sampling and reconstructions of MRI.

❄️ Score-Based Point Cloud Denoising (ICCV 2021)

💭 Diffusion Probabilistic Models for 3D Point Cloud Generation (CVPR 2021)

Doing non-Cartesian MR Imaging has never been so easy.

JAX Arrays for human consumption

Tensors, for human consumption

Point cloud diffusion for 3D model synthesis

Hardware accelerated, batchable and differentiable optimizers in JAX.

Industry-strength Computer Vision workflows with Keras

Do ResNets discretize Neural ODEs?

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

This repo contains the official implementation of HAPPIER: Hierarchical Average Precision Training for Pertinent Image Retrieval (ECCV'22).

Coding assistance for JupyterLab (code navigation + hover suggestions + linters + autocompletion + rename) using Language Server Protocol

Sandbox for training deep learning networks

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

Debug your GitHub Actions via SSH by using tmate to get access to the runner system itself.

Benchmarking for MRI reconstruction using PySap MRI & Benchopt Package

Benchmarking inverse problem solvers on images