| Name | Version | Summary | date |
| jaxutils-nightly |
0.0.8.dev20241206 |
Utility functions for JaxGaussianProcesses |
2024-12-06 00:06:36 |
| jaxkern-nightly |
0.0.5.dev20241206 |
Kernels in Jax. |
2024-12-06 00:05:18 |
| brax |
0.12.0 |
A differentiable physics engine written in JAX. |
2024-12-05 22:09:56 |
| NetKet |
3.15.1 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2024-12-05 21:13:10 |
| transformers |
4.47.0 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-12-05 17:29:36 |
| diffusers-qaic |
0.24.0 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-12-05 15:19:37 |
| easydel |
0.0.80 |
An open-source library to make training faster and more optimized in JAX |
2024-12-04 15:22:10 |
| jax-hpc-profiler |
0.2.9 |
HPC Plotter and profiler for benchmarking data made for JAX |
2024-11-29 19:38:37 |
| matrax |
0.0.5 |
Matrix games in JAX |
2024-11-28 09:02:43 |
| jax-scipy-spatial |
0.2.5 |
Scipy spatial API for JAX |
2024-11-27 16:26:13 |
| jsrm |
0.0.13 |
Kinematic and dynamic models of continuum and articulated soft robots. |
2024-11-22 23:56:38 |
| fjformer |
0.0.91 |
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer in JAX |
2024-11-14 10:11:07 |
| apebench |
0.1.1 |
Benchmark suite for Autoregressive Neural Emulators of PDEs in JAX. |
2024-11-09 08:49:25 |
| gpjax |
0.9.3 |
Gaussian processes in JAX. |
2024-11-01 19:42:49 |
| jax-fdm |
0.8.6 |
Auto-differentiable and hardware-accelerated force density method |
2024-10-30 19:25:13 |
| jax-fem |
0.0.8 |
A GPU accelerated Finite element analysis package in JAX. |
2024-10-30 02:46:50 |
| jax-flash-attn2 |
0.0.1 |
Flash Attention Implementation with Multiple Backend Support and Sharding This module provides a flexible implementation of Flash Attention with support for different backends (GPU, TPU, CPU) and platforms (Triton, Pallas, JAX). |
2024-10-23 22:37:19 |
| exponax |
0.1.0 |
Efficient differentiable PDE solvers in JAX. |
2024-10-23 07:30:54 |
| diffusers |
0.31.0 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-10-22 14:13:23 |
| flowjax |
16.0.0 |
Easy to use distributions, bijections and normalizing flows in JAX. |
2024-10-17 11:00:06 |