vmas
VMAS is a vectorized differentiable simulator designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of a vectorized 2D physics engine written in PyTorch and a set of challenging multi-robot scenarios. Additional scenarios can be implemented through a simple and modular interface.
ragoon
High level library for batched embeddings generation, blazingly-fast web-based RAG and quantized indexes processing ⚡
tdarec
recipes + dials extension for persistent homology and vectorizations thereof
math-array-tools-unary
Constructor for applying a unary function to each element in an input array.
katmer
katmer is a powerful library for optimizing the design of optical thin films using automatic differentiation via JAX and Equinox, enabling efficient and accurate inverse design solutions.
kernel_float
CUDA/HIP header-only library for low-precision (16 bit, 8 bit) and vectorized GPU kernel development
https://github.com/bahremsd/tmmax-workshop
Workshop given in graduate-level thin film coatings course in ITU
https://github.com/cimagroup/vectorization-maps
tmmax
A fast transfer matrix method written in jax for modelling optical multilayer thin films