Jax
Showing 5 of 113 projects
A photovoltaic simulator with automatic differentiation for solar cell modeling and optimization, built on JAX.
A JAX-based library for loopy belief propagation on discrete factor graphs, enabling efficient probabilistic inference.
A differentiable hydrodynamics and magnetohydrodynamics code for astrophysics built with JAX, enabling gradient-based inverse modeling and multi-GPU simulations.
GPU/TPU accelerated nonlinear least-squares curve fitting using JAX, designed as a drop-in replacement for SciPy's curve_fit.
A differentiable ray tracing toolbox for radio propagation simulations, built on JAX for optimization and machine learning.
Related Tags
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.