Open-source alternatives to Jaxdf

tmmax is a high-performance computational library designed for efficient calculation of optical properties in multilayer thin-film structures. Engineered to serve as a Swiss Army knife tool for thin-film optics research, tmmax integrates a comprehensive suite of advanced numerical methods. At…

Biopython is a bioinformatics library for Python providing tools to parse, manipulate, and analyze biological sequences, molecular structures, and phylogenetic trees. It serves as a biological sequence parser for genomic and proteomic data across multiple industry-standard file formats and acts as an interface for querying biological data and citations from NCBI Entrez repositories. The project distinguishes itself through specialized toolkits for protein structure analysis and phylogenetic tree construction. It includes a protein structure analyzer for processing PDB and mmCIF files to calcu

Brain Dynamics Programming in Python

API reference PyPI

.. docs-include-ref

Functional Models and Algorithms for Sparse Signal Processing

Parsers and algorithms for computational chemistry logfiles

A State-based Transformation System for Brain Modeling.

Leveraging Taichi Lang to customize brain dynamics operators.

Physical units and unit-aware math system for general-purpose brain dynamics modeling

Biologically Detailed Brain Cell Modeling in JAX

Colour Science for Python

High-performance quantum systems simulation with JAX (GPU-accelerated & differentiable solvers).

This is a V module to create neural networks. (there is also a gui module named ggui that I used for the mnist example)

vplot is a wrapper for GNU Plot (gnuploti). The source of gnuploti I have downloaded from this link. Files listed on gnuplot_i are taken from downloaded sources, including the license file.

econpizza .. |badge0| image:: https://img.shields.io/badge/GitHub-gboehl%2Feconpizza-blue.svg?style=flat :target: https://github.com/gboehl/econpizza .. |badge1| image:: https://github.com/gboehl/econpizza/actions/workflows/continuous-integration.yml/badge.svg :target:…

Quickstart | Reference docs | Paper | NeurIPS 2020

See full documentation for detailed info.

VIVSIM is a Python library for fluid-structure interaction (FSI) simulations based on the immersed boundary -lattice Boltzmann method (IB-LBM). It was originated from a research project requiring efficient simulation codes for studying vortex-induced vibration (VIV) of underwater structures.

Differentiable spectral modelling of exoplanets/brown dwarfs/M dwarfs using JAX! Read the docs 🐕 or deepwiki for ExoJAX.

Foragax is an Agent-Based Modelling (ABM) package based on JAX. It provides scalable and efficient ABM simulations by leveraging JAX's automatic vectorization and just-in-time compilation capabilities. The main features of Foragax include:

Differentiable Ray Tracing Toolbox for Radio Propagation Simulations

SCICO is a Python package for solving the inverse problems that arise in scientific imaging applications. Its primary focus is providing methods for solving ill-posed inverse problems by using an appropriate prior model of the reconstruction space. SCICO includes a growing suite of operators,…

astronomix (formerly jf1uids) is a differentiable hydrodynamics and magnetohydrodynamics code written in JAX with a focus on astrophysical applications. astronomix is easy to use, well-suited for fast method development, scales to multiple GPUs and its differentiability opens the door for…

Manim is a scriptable, code-driven framework designed for generating precise technical visualizations and mathematical animations. By using a high-level programming interface, it allows users to define geometric shapes, motion paths, and animation logic that are compiled into high-quality video assets. The system functions as a specialized engine for creating reproducible, data-driven representations of complex mathematical concepts and geometric transformations. The framework distinguishes itself through an interpolation-based engine that calculates intermediate states between keyframes to e

Mesa is a Python framework for agent-based modeling and complex systems simulation. It provides a toolkit for creating simulations where autonomous agents interact within a shared environment to observe emergent global behaviors. The project includes a browser-based interface for real-time visualization of simulation states and agent interactions. It also functions as a data analysis library for recording and processing metrics from model runs to quantify system behavior. The framework supports multiple environmental topologies, including grid-based spatial mapping and network-graph topologi

NetKet is an open-source project delivering cutting-edge methods for the study of many-body quantum systems with artificial neural networks and machine learning techniques. It is a Python library built on JAX.