Starred repositories
JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
Interpolation and function approximation with JAX
Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.
Easily serialize Data Classes to and from JSON
Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.
Javascript Canvas Library, SVG-to-Canvas (& canvas-to-SVG) Parser
AI Hilbert is an algebraic geometric based discovery system (based on Putinar's Positivstellensatz), that enables the discovery of fundamental laws of nature (or non-physical systems) based on know…
The Temperature Dependent Effective Potentials (TDEP) code
An Agile RISC-V SoC Design Framework with in-order cores, out-of-order cores, accelerators, and more
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
A GPU-accelerated, differentiable, Maxwell FDTD solver
An 808-line Matlab educational code for combined multi-scale topology optimisation and phasor-based dehomogenisation
Efficient forward- and reverse-mode sparse Jacobians using Jax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Orbax provides common checkpointing and persistence utilities for JAX users
Multiple dispatch over abstract array types in JAX.
A JupyterLab extension for collaborative 3D geometry modeling
ICRAR / ijson
Forked from isagalaev/ijsonIterative JSON parser with Pythonic interfaces
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
A plotly.js React component from Plotly 📈