GoMLX: An Accelerated Machine Learning Framework For Go
-
Updated
Dec 17, 2025 - Go
#
xla
Here are 65 public repositories matching this topic...
Easy to use and blazing fast JAX-based library for high-performance 2D/3D Discrete Element Method (DEM) simulations.
simulation hpc gpu physics molecular-dynamics ml physics-engine python3 scientific-computing rl physics-simulation md discrete-element-method tpu jax xla
-
Updated
Dec 17, 2025 - Python
S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.
deep-learning simulation optimization autograd simulation-framework simulations circuit physics-simulation ann photonics s-parameters jax xla photonic-circuit photonic-optimization
-
Updated
Dec 16, 2025 - Python
Elegant and Performant Deep Learning
-
Updated
Dec 15, 2025 - Julia
Enabling PyTorch on XLA Devices (e.g. Google TPU)
-
Updated
Dec 10, 2025 - C++
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡
-
Updated
Nov 19, 2025 - Python
Repository of Computer Vision projects based on CNNs, Vision Transformers, and YOLO11, implemented with TensorFlow, PyTorch, Hugging Face, and Ultralytics.
python opencv computer-vision tensorflow cnn pytorch image-classification object-detection image-segmentation multi-gpu pose-estimation tpu xla distributed-data-parallel vision-transformer hugging-face ultralytics yolo11
-
Updated
Nov 10, 2025
python opencv tensorflow plotly cnn pandas pytorch human-pose-estimation multi-gpu pose-estimation hand-pose-estimation tpu xla animal-pose-estimation distributed-data-parallel convnext yolo11 face-landmark-estimation
-
Updated
Nov 10, 2025 - Jupyter Notebook
A fast transfer matrix method written in jax for modelling optical multilayer thin films
jit optics vectorization photonics thin-film coating tmm jax transfer-matrix-method xla optical-coatings optical-simulation thin-film-engineering jax-based-optical-tools optical-coating
-
Updated
Oct 27, 2025 - Jupyter Notebook
Experimentation using the xla compiler from rust
-
Updated
Oct 18, 2025 - Rust
JAX Foreign Function Interface experiments
-
Updated
Oct 7, 2025 - C++
JAX - A curated list of resources https://github.com/google/jax
-
Updated
Sep 2, 2025
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.
simulation optics sensitivity vectorization equinox photonics thin-film tmm jax xla inverse-design coatings tmmax
-
Updated
Aug 18, 2025 - Python
Improve this page
Add a description, image, and links to the xla topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the xla topic, visit your repo's landing page and select "manage topics."