arpastrana / jax_fdm
Auto-differentiable and hardware-accelerated force density method
☆83Updated this week
Related projects: ⓘ
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆48Updated 3 weeks ago
- A differentiable finite element analysis solver for structural optimization based on JAX☆20Updated 2 weeks ago
- ☆32Updated last year
- Differentiable interface to FEniCS for JAX☆49Updated 3 years ago
- XLB: Accelerated Lattice Boltzmann (XLB) based on JAX for Physics-based ML☆188Updated last week
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆87Updated 10 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆41Updated last week
- PyTorch-FEniCS interface☆97Updated 3 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆47Updated 7 months ago
- A shape optimisation toolbox built on the Firedrake finite element library.☆26Updated last week
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆146Updated last year
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆66Updated 3 months ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆55Updated 2 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆119Updated this week
- A Mesh-free Topology Optimization Method using Neural Representations☆27Updated 2 years ago
- Neural reparameterization improves structural optimization☆117Updated 3 years ago
- Differentiable Finite Element Method with JAX☆235Updated 2 weeks ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆28Updated 7 months ago
- DL4TO is a Python library for 3D topology optimization that is based on PyTorch and allows easy integration with neural networks.☆42Updated last month
- Code, technical details and links to datasets and trained models for TopoDiff☆33Updated 10 months ago
- ☆11Updated 3 years ago
- A Python Library for Topology Optimization☆94Updated 2 years ago
- ☆51Updated this week
- ☆16Updated last year
- Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.☆15Updated 10 months ago
- The algorithmic differentation tool pyadjoint and add-ons.☆90Updated last week
- Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"☆35Updated last year
- Python project dedicated to creating an open-source CAD designer using implicit equations.☆32Updated 7 months ago
- Compute interesting points, areas, and volumes in simplex meshes of any dimension.☆104Updated 3 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆34Updated last month