Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation

A repository for quantum chemistry basis sets

A Python package for probabilistic state space modeling with JAX

Coordinates in JAX

Interpolation and function approximation with JAX

Thermodynamic Hypergraphical Model Library in JAX

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

AMReX-based structured wind solver

Athena++ radiation GRMHD code and adaptive mesh refinement (AMR) framework

Unofficial reference for the NHL API endpoints.

Kolmogorov Arnold Networks

jax for quantum simulations

A curated list of awesome high performance computing resources

Automatic differentiation tools for interfacing custom density functionals with libXC.

For when people get too hyped up about things

🌱 a fast, batteries-included static-site generator that transforms Markdown content into fully functional websites

The open source AI app collection

A generative model for programmable protein design

GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine learning techniques.

Formalizing results about the Mandelbrot set in Lean

🐧 Penguins in LaTeX with TikZ

Computational algebraic number theory

Cataclysm - Code generation library for the end game

A JIT compiler for hybrid quantum programs in PennyLane

Implementation for SE(3) diffusion model with application to protein backbone generation

jax library for E3 Equivariant Neural Networks

Chemical reaction network and systems biology interface for scientific machine learning (SciML). High performance, GPU-parallelized, and O(1) solvers in open source software.

Development repository for the Triton language and compiler

A massively parallel, optimal functional runtime in Rust