tumaer/JAXFLUIDS
Differentiable Fluid Dynamics Package
What it solves
JAX-Fluids는 머신러닝과 전산 유동역학(CFD) 사이의 간극극을 메우기 위해 설계되었습니다. 3D 압축성 단상 및 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 args 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 Possibility 兩 args 兩 研究- 兩 args 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 ing 兩 兩 兩
How it works
JAX로 완전히 구축된 이 패키지는 데카르트 그리드에서 유한체적법을 사용하여 Navier-Stokes 방정식을 풉니다. WENO và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và và 兩 兩 兩 兩 兩 兩
Who it's for
ML과 CFD의 접점에 있는 연구자들을 위한 것입니다. CPU, GPU, 그리고 TPU hardware への への への 대- への 兩 への への への への への への への への への への への への への への への への への への への への への への への への への への への への への への への への への への への ing 兩 への への 兩 への への への への への への への への への への への への への への 대- への への への への への への への への への への への への への への への への への への への への への F への ................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................
Highlights
- Fully Differentiable: 모델 최적화를 위한 자동 미분을 가능하게 합니다.
- **n/a (註：原文- 在 CPU, GPU, 與 TPU 上運行。)
- Versatile Flow Models:
- Hardware Agnostic: CPU, GPU, 및 TPU에서 실행됩니다.
0 0 兩 兩 兩 兩 兩 兩 兩 n/a (註 0 兩 兩 兩 n/A/A/A/A/A/A/A/A/A/A/A/J/A/A/A/A/A/A/A/A/A/A/args 兩 A/A/................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................: 0 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 0 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 への への への への への への への への への 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩 兩................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................ |} ] } (註：原文- 在 CPU, GPU, 與 TPU 上運行。)
Highlights
- Fully Differentiable: 모델 최적화를 위한 자동 미분을 가능하게 합니다.
- High Performance: HPC 集群上高效擴展, 已在多達 512 NVIDIA A100 GPUs 與 2048 TPU-v3 cores 上進行過測試。
- Versatile Flow Models: 支援 3D 可壓縮流體、兩相模擬與浸入式固體邊界。
- Hardware Agnostic: 在 CPU, GPU, 與 TPU 上運行。