-
-
fdtdx Public
Electromagnetic FDTD Simulations in JAX
-
-
-
interpax Public
Forked from f0uriest/interpaxInterpolation and function approximation with JAX
Python MIT License UpdatedOct 17, 2025 -
awesome-jax Public
Forked from n2cholas/awesome-jaxJAX - A curated list of resources https://github.com/google/jax
Creative Commons Zero v1.0 Universal UpdatedSep 2, 2025 -
jax Public
Forked from jax-ml/jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python Apache License 2.0 UpdatedAug 27, 2025 -
-
-
-
-
-
annax Public
Forked from atksh/annaxannax is an ANN library based on JAX that provides fast and memory-efficient approximate nearest neighbor search for high-dimensional data. It includes a wide range of indexing structures and dista…
Python MIT License UpdatedJun 3, 2025 -
pytreeclass Public
Forked from ASEM000/pytreeclassVisualize, create, and operate on pytrees in the most intuitive way possible.
-
quax Public
Forked from patrick-kidger/quaxMultiple dispatch over abstract array types in JAX.
Python Apache License 2.0 UpdatedJul 20, 2024 -
-
subgraphx Public
Reproducing results from the SubgraphX paper including a full re-implementation of the SubgraphX algorithm in PyTorch-Geometric
-
ppo_compiler_gym Public
Submission of PPO Implementation for Compiler Gym
-
lie_learn Public
Forked from AMLab-Amsterdam/lie_learnComputations involving Lie groups and harmonic analysis
Python MIT License UpdatedApr 7, 2023 -
-
-
CompilerGym Public
Forked from facebookresearch/CompilerGymReinforcement learning environments for compiler and program optimization tasks
Python MIT License UpdatedFeb 22, 2022 -
{{ message }}