19 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Large-scale generative pretrain of single cell using transformer. | 4K | |
| A versatile and easy to understand Diffusion library | 3K | |
| zlynx is a lightweight, modular library built on top of JAX and Flax NNX for imp... | 2K | |
| Hax - A suite for Heterogeneity analysis for CryoEM data | 1K | |
| Architecture experimentation and training infrastructure. | 1K | |
| JAX implementation of Reinforcement Learning agents | 1K | |
| Modeling complex perturbations with CellFlow | 808 | |
| Towards Embodied AI with MuscleMimic: Unlocking full-body musculoskeletal motor ... | 485 | |
| Felafax is building AI infra for non-NVIDIA GPUs | 451 | |
| 375 | ||
| LEM slope stability analysis using JAX | 334 | |
| JAX-LNN (JLNN) | 294 | |
| Imitation learning benchmark focusing on complex locomotion tasks using MuJoCo. | 287 | |
| Bayesian Data Analysis using PyMC and JAX | 255 | |
| GPU-accelerated MDP solvers in Python with JAX | 223 | |
| A Graph Neural Network library for real-life energy networks. | 165 | |
| 91 | ||
| Harnessing the Power of Single Cell Large Language Models with Parameter Efficie... | 87 | |
| Add your description here | 6 |
