83 Open source projects that are alternatives of or similar to jax-cfd

Repository for the open-source lethe CFD/DEM/CFD-DEM project

Solver for wind farm simulations targeting exascale computational platforms

This is a replication of "CFD Python: 12 steps to Navier-Stokes" in Julia

GitHub extension for agile project management, using the issues subsystem.

Python code for "Machine learning: a probabilistic perspective" (2nd edition)

Computational Fluid Dynamics based on PyTorch and the Lattice Boltzmann Method

Unstructured Compressible Navier Stokes 3D code (UCNS3D)

A sequence of Jupyter notebooks featuring the "12 Steps to Navier-Stokes" http://lorenabarba.com/

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

A low Mach number stellar hydrodynamics code

Toolbox to simulate GNF and viscoelastic fluid flows in OpenFOAM®

Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)

FOSolverS: a Suite of Free & Open Source Solvers. A brief overview:

Python classes for easier creation of OpenFOAM's blockMesh dictionaries.

A versatile Python-Fortran CFD code that solves a large class of 2D flows

An Open Source Forex Trading Platform

Materials for training tracks for continua media - OpenFOAM, vortex method, and other

MacroUtils is a collection of high-level APIs in order to make your life easier when writing STAR-CCM+ JAVA macros.

A code for fast, massively-parallel direct numerical simulations (DNS) of canonical flows

Deep learning operations reinvented (for pytorch, tensorflow, jax and others)

Simple 2d finite volume solver for Euler equations using c++ kokkos library

Flax is a neural network library for JAX that is designed for flexibility.

Basic Computational Fluid Dynamics (CFD) schemes implemented in FORTRAN using Finite-Volume and Finite-Difference Methods. Sample simulations and figures are provided.

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

DLBFoam: An open-source dynamic load balancing model for fast reacting flow simulations in OpenFOAM. https://doi.org/10.1016/j.cpc.2021.108073

No description or website provided.

Implementation of https://srush.github.io/annotated-s4

Two-dimensional flow solver with GUI using vortex particle and boundary element methods

Generate statistics from Taiga and produce burnup diagrams, CFDs, dependency graphs and more.

Graphsignal Python agent

Universal modeling and simulation of fluid dynamics upon machine learning

Object-oriented multi-mesh version of the classic reacting turbulent multiphase flow solver

eddylicious.readthedocs.io/

ADAM implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.

No description or website provided.

DAFoam: Discrete Adjoint with OpenFOAM for High-fidelity Gradient-based Design Optimization

Efficient computations with symmetric and non-symmetric tensors with support for automatic differentiation.

MagIC is a high-performance code that solves the magneto-hydrodynamics equations in rotating spherical shells

OpenFoam postprocessing python tool

Computational fluid dynamics codes

Range Software - Finite Element Analysis

The fast Finite Volume simulator with UQ support.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

United collection of hybrid Central solvers - one-phase, two-phase and multicomponent versions

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

A multi-block solver for massively parallel direct numerical simulations (DNS) of fluid flows

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

2D solver for Euler equations in quadrilateral grid, using kinetic flux vector splitting scheme, written in OOP F2003

Trax — Deep Learning with Clear Code and Speed

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

FEATool - "Physics Simulation Made Easy" (Fully Integrated FEA, FEniCS, OpenFOAM, SU2 Solver GUI & Multi-Physics Simulation Platform)

A small library for creating and manipulating custom JAX Pytree classes

2D orthogonal elliptic mesh generator which solves the Winslow partial differential equations

IG Markets API wrapper for Node.js

JAX implementations of core Deep RL algorithms

Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.

A 2D finite volume CFD code, written in C

Wavelet Adaptive Block-Based solver for Interactions with Turbulence

ExaDG - High-Order Discontinuous Galerkin for the Exa-Scale