Tool for live presentations using manim

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

Production-grade 3D gaussian splatting with CPU/GPU support for Windows, Mac and Linux 🚀

PyTorch implementation of normalizing flow models

A simple & elegant experiment tracking framework that integrates persistence logic & best practices directly into Python

WebGPU for Python

Physics-Informed Neural Networks (PINN) Solvers of (Partial) Differential Equations for Scientific Machine Learning (SciML) accelerated simulation

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

A PyTorch implementation of "Graph Wavelet Neural Network" (ICLR 2019)

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

Scripts and resources to create Hetionet v1.0, a heterogeneous network for drug repurposing

Generating expression signatures for disease using STARGEO

Python Socket.IO server and client

Your every day Keras toolbox.