Awesome Open Source

Programming Languages

Search results for scientific computing jax

scientific-computing x

6 search results found

Pdebench ⭐ 522

PDEBench: An Extensive Benchmark for Scientific Machine Learning

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

Probdiffeq ⭐ 25

Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and custom information operators. Compatible with the broader JAX scientific computing ecosystem.

Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.

Diffeqzoo ⭐ 7

A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiological models, IVPs, BVPs, and more.

Neural Cluster Update ⭐ 7

Code for 'Unbiased Monte Carlo Cluster Updates with Autoregressive Neural Networks'.

1-6 of 6 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.