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

CUDA on AMD 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

Machine Learning Engineering Open Book

Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*

3D plotting and mesh analysis through a streamlined interface for the Visualization Toolkit (VTK)

Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.

FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores