aadityacs

jaxtyping

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

holoviews

With Holoviews, your data visualizes itself.

neural-isometries

Official JAX implementation of neural isometries - taming transformations for equivariant ML

leaderboard

Database of solutions to inverse design challenges

PhoTOS

Topology Optimization of Photonic Components using a Shape Library

tree-diffusion

Diffusion on syntax trees for program synthesis

python-ternary

:small_red_triangle: Ternary plotting library for python with matplotlib

GPJax

Gaussian processes in JAX.

colour

Python color representations manipulation library (RGB, HSL, web, ...)

VoroTO

Multiscale Topology Optimization of Voronoi Structures using Surrogate Neural Networks

siren-flax

SIREN neural networks in Flax

sparsejac

Efficient forward- and reverse-mode sparse Jacobians using Jax

optimism

Computational solid mechanics made easy with Jax

optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

PlotNeuralNet

Latex code for making neural networks diagrams

BSP-NET-pytorch

PyTorch 1.2 implementation of BSP-NET.

NeuRBF

PtychoNN

Deep learning of ptychographic imaging

jax-fem

Differentiable Finite Element Method with JAX

PtychoPINN

fast, high-resolution lensless imaging in Tensorflow

TopOpt-ThermoElastic

Thermoelastic topology optimization procedure considering steady-state heat conduction

awesome-implicit-representations

A curated list of resources on implicit neural representations.

gym

A collection of inverse design challenges

cmcrameri

Making Fabio Crameri's perceptually uniform colourmaps for geosciences available on PyPI and conda-forge

spherical-cnn

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

jax-fenics-adjoint

Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint

point-cloud-utils

An easy-to-use Python library for processing and manipulating 3D point clouds and meshes.

Open3D

Open3D: A Modern Library for 3D Data Processing

cola

Compositional Linear Algebra