TORAX: Tokamak transport simulation in JAX

Codebase for PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs.

Agustinus' very opiniated publication-ready plotting library

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

Dr.Jit — A Just-In-Time-Compiler for Differentiable Rendering

Torchhd is a Python library for Hyperdimensional Computing and Vector Symbolic Architectures

Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff

Multithreaded version of numpy.einsum

🔍 finite element analysis for continuum mechanics of solid bodies

Universal Tensor Operations in Einstein-Inspired Notation for Python.

ManimML is a project focused on providing animations and visualizations of common machine learning concepts with the Manim Community Library.

A CFD open source code dedicated to multiphase compressible flows

Parallel High-Order Library for PDEs through hp-adaptive Discontinuous Galerkin methods

Dataframes powered by a multithreaded, vectorized query engine, written in Rust

Cross-platform, fast, feature-rich, GPU based terminal

A conda-smithy repository for jax.

An implementation of chunked, compressed, N-dimensional arrays for Python.

Links to works on deep learning algorithms for physics problems, TUM-I15 and beyond

Python bindings for the Transformer models implemented in C/C++ using GGML library.

Program for Simulation of Turbulent Flows

Code at the speed of thought – Zed is a high-performance, multiplayer code editor from the creators of Atom and Tree-sitter.

The open and composable observability and data visualization platform. Visualize metrics, logs, and traces from multiple sources like Prometheus, Loki, Elasticsearch, InfluxDB, Postgres and many mo…

FLEXI: A high order discontinuous Galerkin framework for hyperbolic–parabolic conservation laws

[ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"

CUDA on ??? GPUs

The Foundation for All Legate Libraries

An Aspiring Drop-In Replacement for NumPy at Scale

Python package for numerical derivatives and partial differential equations in any number of dimensions.

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

jax-triton contains integrations between JAX and OpenAI Triton