parallax

MIT License

Stars
157
View Code on GitHub View on X
Ecosystems: Python

Statistics for this project are still being loaded, please check back later.

Related Projects
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
serket

The ✨Magical✨ JAX ML Library.

20 Aug 2022 16
Lightweighting_Cookbook

This project attempts to build neural network training and lightweighting cookbook including thre...

25 May 2022 18
PyTorch-Reparam-Module

Reparameterize your PyTorch modules

22 Oct 2018 69
minGPT-flax

GPT implementation in Flax

30 Sep 2021 16
torchinfo

View model summaries in PyTorch!

16 Mar 2020 2,523
lpips-j

Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hug...

08 Jul 2022 14
jax2torch

Use Jax functions in Pytorch

26 Oct 2021 224
diffusion-transformer

Pytorch and JAX Implementation of Scalable Diffusion Models with Transformers | Diffusion Transfo...

16 Aug 2024 2
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
jraph

A Graph Neural Network Library in Jax

23 Nov 2020 1,327
ninjax

General Modules for JAX

23 May 2022 57
Neural_Net_Using_NumPy

Feed Forward Neural Networks using Numpy

24 Apr 2020 1
lab

A generic interface for linear algebra backends

29 Mar 2018 67
dm-haiku

JAX-based neural network library

18 Feb 2020 2,816