kasper0406

kasper0406's Stars

• iree-org/iree

  A retargetable MLIR-based machine learning compiler and runtime toolkit.

  Language:C++2.9k640

• eiln/ane

  Reverse engineered Linux driver for the Apple Neural Engine (ANE).

  Language:C38318

• google-deepmind/jraph

  A Graph Neural Network Library in Jax

  Language:Python1.4k91

• blackjax-devs/blackjax

  BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

  Language:Python870108

• openxla/xla

  A machine learning compiler for GPUs, CPUs, and ML accelerators

  Language:C++2.8k472

• tianjuxue/jax-am

  Additive manufacturing simulation with JAX.

  Language:Jupyter Notebook28156

• deepmodeling/jax-fem

  Differentiable Finite Element Method with JAX

  Language:Python32648

• NVIDIA/warp

  A Python framework for high performance GPU simulation and graphics

  Language:Python4.4k255

• Autodesk/XLB

  XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML

  Language:Python25228

• google/brax

  Massively parallel rigidbody physics simulation on accelerator hardware.

  Language:Jupyter Notebook2.5k265

• google/flax

  Flax is a neural network library for JAX that is designed for flexibility.

  Language:Jupyter Notebook6.3k659

• patrick-kidger/diffrax

  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

  Language:Python1.5k140

• jax-ml/jax

  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  Language:Python31k2.9k

• apple/coremltools

  Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

  Language:Python4.5k661

• google-deepmind/optax

  Optax is a gradient processing and optimization library for JAX.

  Language:Python1.8k201

• Gallopsled/pwntools

  CTF framework and exploit development library

  Language:Python12.3k1.7k